pytorch:tensorをtorch.catしたい
pytorchでプログラムを書いているのですが、torch.catができません。
以下のようなエラーが発生しています。
エラーメッセージ
TypeError: expected Variable as element 0 in argument 0, but got list
###ソースコード
p = [] for i in range(0, N, self.gpu_batch): p.append(self.model(stack[i:min(i + self.gpu_batch, N)])) p = torch.cat(p) # Number of classes CL = p.size(1) sal = torch.matmul(p.data.transpose(0, 1), self.masks.view(N, H * W)) sal = sal.view((CL, H, W)) sal = sal / N / self.p1 return sal
解決方法があれば、ご教授頂きたいです。
よろしくお願いします。
補足情報
pytorchのバージョンは0.4.0です。
あなたの回答
tips
プレビュー