pytorch LSTM图像分类

一个问题:pytorch官方文档对LSTM的输入参数的格式是input of shape (seq_len, batch, input_size),可是本例中images.reshape(-1, sequence_length, input_size)的输入格式为batch,seq_len, input_size,是否是官文写错了? import torch import torch.nn as
相关文章
相关标签/搜索