RNN 差异对比

torch.nn.LSTM

torch.nn.LSTM(
   input_size,
   hidden_size,
   num_layers=1,
   bias=True,
   batch_first=False
   dropout=0,
   bidirectional=False,
   proj_size=0
)

更多请查看 torch.nn.LSTM.

megengine.module.LSTM

  megengine.module.LSTM(
     input_size,
     hidden_size,
     num_layers=1,
     bias=True,
     batch_first=False
     dropout=0,
     bidirectional=False,
     proj_size=0
)

更多请查看 megengine.module.LSTM.

参数与使用方式基本一致。