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
.
参数与使用方式基本一致。