TypeError:view()最多接受2个参数(给定3个)

时间:2019-04-23 06:08:40

标签: lstm pytorch torch

我尝试在pytorch中使用view(),但是我无法输入3个参数。我不知道为什么它会不断出现此错误?谁能帮我这个?

    def forward(self, input):
        lstm_out, self.hidden = self.lstm(input.view(len(input), self.batch_size, -1))

1 个答案:

答案 0 :(得分:0)

您的input似乎是一个numpy数组,而不是割炬张量。您需要先进行转换,例如input = torch.Tensor(input)