pytorchでtorch.catができない

Question

前提・実現したいこと

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 += [self.model(stack[i:min(i + self.gpu_batch, N)])]
        p = torch.cat(p)

試したこと

listをnumpyにしてからtensorにしてからtorch.catできるか試しましたがうまくいきませんでした。

補足情報（FW/ツールのバージョンなど）

torch:0.4.0を使用しています。

Answer 1

これ

p += [self.model(stack[i:min(i + self.gpu_batch, N)])]

をこう

p += self.model(stack[i:min(i + self.gpu_batch, N)])

すると解決できる気がします。