PyTorchのnn.Sequentialを利用する方法は何ですか?

PyTorchのnn.Sequentialは、ニューラルネットワークモデルを構築するためのコンテナです。複数の層を順番に組み合わせて、ディープラーニングモデルを構築することができます。

nn.Sequentialの使い方は以下のようです:

import torch.nn as nn

model = nn.Sequential(
    nn.Linear(input_size, hidden_size),  # 添加输入层
    nn.ReLU(),                           # 添加激活函数
    nn.Linear(hidden_size, output_size),  # 添加输出层
    nn.Sigmoid()                         # 添加激活函数
)

上記の例では、nn.Sequentialは、順番に4つの層を追加しました:線形層、ReLU活性化関数、別の線形層、そしてSigmoid活性化関数です。これらの層は、追加された順序で入力データに順番に適用され、最終的な出力が生成されます。

このnn.Sequentialモデルを使用するには、model(x)を呼び出すことができます。ここで、xは入力データです。モデルは、入力データを追加された各層に順次渡し、最終的な出力を生成します。

bannerAds