nn.Linear 对输入的tensor张量格式要求
时间: 2023-11-05 19:37:40 浏览: 114
nn.Linear 对输入的 tensor 张量格式要求是二维张量,即形状为 (batch_size, input_size) 的张量。其中,batch_size 表示输入的样本数量,input_size 表示每个样本的特征维度。如果输入的张量不是二维张量,nn.Linear 会抛出异常。如果输入的张量形状不符合要求,可以使用 reshape 函数将其转换为二维张量,或者使用其他的层对输入进行预处理。
阅读全文