pytorch源碼解讀——RNN/LSTM篇

文章的字母中:python b: batch_size t: time_step n: num_feature h: hidden_size 假設輸入數據維度input = (b, t, n) 所設計的LSTM模型以下:web class MYLSTM(nn.Module): def __init__(self, input_size, hidden_size, out_size):
相關文章
相關標籤/搜索