1
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 1 year has passed since last update.

2023/3現在 PyTorchのLSTM引数についてまとめる

Posted at

input_size
入力系列データの変数の数

hidden_size
隠れ状態の変数の数

num_layers
LSTMユニットを何層重ねるか(デフォルト:1)

bias
Falseに設定するとb_ihとb_hhを使わなくなる(デフォルト:True)

batch_first
Trueに設定すると入力するTensorの形を (データ長, バッチサイズ, 特徴量)から(バッチサイズ, データ長, 特徴量)に変更できる,つかうひとのこのみ(デフォルト:False)

dropout
最後のLSTM層を除いて設定した値でDropoutする(デフォルト:0)

bidirectional
Trueに設定すると双方向LSTMに変更できる(デフォルト:False)

proj_size
正の値を設定すると対応したサイズのprojection layerを使う,おそらくLSTMPになる(デフォルト:0)

参考
https://runebook.dev/ja/docs/pytorch/generated/torch.nn.lstm

1
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?