RNN 差异对比

torch.nn.LSTMCell

torch.nn.LSTM(
   input_size,
   hidden_size,
   bias=True
)

:

更多请查看 torch.nn.LSTMCell.

—

  megengine.module.LSTMCell(
     input_size,
     hidden_size,
     bias=True
)

更多请查看 megengine.module.LSTMCell.

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