RNN 差异对比

torch.nn.RNNCell

torch.nn.RNNCell(
   input_size,
   hidden_size,
   bias=True,
   nonlinearity='tanh',
   device=None,
   dtype=None
)

更多请查看 torch.nn.RNNCell.

megengine.module.RNNCell

  megengine.module.RNNCell(
     input_size,
     hidden_size,
     bias=True,
     nonlinearity='tanh'
)

更多请查看 megengine.module.RNNCell.

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