在Torch中定義一個簡單的神經網絡模型可以使用nn模塊。以下是一個簡單的神經網絡模型定義示例:
require 'nn'
-- 定義一個簡單的神經網絡模型
model = nn.Sequential()
model:add(nn.Linear(10, 20)) -- 輸入層到隱藏層的線性變換,輸入10個神經元,輸出20個神經元
model:add(nn.ReLU()) -- 激活函數,此處使用ReLU
model:add(nn.Linear(20, 1)) -- 隱藏層到輸出層的線性變換,輸入20個神經元,輸出1個神經元
-- 打印模型結構
print(model)
在這個示例中,我們定義了一個簡單的神經網絡模型,包括一個包含兩個線性層和一個ReLU激活函數的神經網絡。我們首先使用nn.Sequential()
創建一個序列化的神經網絡模型,然后使用add()
方法逐層添加神經網絡層。最后,我們打印模型結構,以查看各層的連接關系。
您可以根據自己的需求進一步調整和定制神經網絡模型的結構和參數。