nn.LSTM
PyTorch LSTM API文档
?...输入数据格式:
input:[seq_len, batch, input_size]
$h_0$:[num_layers * num_directions, batch, hidden_size]
$c..._0$:[num_layers * num_directions, batch, hidden_size]
输出数据格式:
output:[seq_len, batch, hidden_size * num_directions...= nn.LSTM(input_size=100, hidden_size=20, num_layers=4)
x = torch.randn(10, 3, 100) # 一个句子10个单词,送进去3..., 20]) torch.Size([4, 3, 20]) torch.Size([4, 3, 20])
nn.LSTMCell
PyTorch LSTMCell API文档
?