PyTorch LSTMCell API文档. 和RNNCell类似,输入input_size的shape是 [batch, input_size] ,输出$h_t$和$c_t$的shape是 [batch, hidden_size].
確定! 回上一頁