pythonのnn.linearの使い方はどうですか。

PyTorchのnn.Linearは、線形変換を定義するためのクラスです。これを使用して全結合層(または線形層とも呼ばれる)を定義することができます。

nn.Linearを使用する例の方法は次のとおりです:

最初に、必要なモジュールをインポートします。

import torch
import torch.nn as nn

次に、入力と出力のサイズを含む線形層を定義します。

input_size = 10
output_size = 5

linear_layer = nn.Linear(input_size, output_size)

入力次元が10の特徴を出力次元が5の特徴にマッピングする線形層が作成されます。

その後、データをリニア層に渡して変換することができます。

input_data = torch.randn(1, input_size)
output_data = linear_layer(input_data)

この機能は、入力データを線形変換し、重みとバイアスに基づいて出力データを返します。

最後に、線形層の重みとバイアスを確認することができます。

print(linear_layer.weight)
print(linear_layer.bias)

線形層の重み行列とバイアスベクトルが印刷されます。

注意:nn.Linearクラスには、デフォルトでバイアスを追加するかどうかや重みの初期化方法など、さまざまな追加パラメータを指定できます。詳細情報はPyTorchの公式ドキュメントを参照してください。

bannerAds