pytorch中列表、numpy混合使用计算的坑

torch中只有cpu中tensor的类型才可与numpy进行混合使用计算，gpu的不行

import torch
import numpy as np

正常输出

a=torch.Tensor([1])
b=torch.Tensor([1])
c=[]
c.append(a)
c.append(b)

np.mean(c)

1.0
============================================================================================================

以下报错:

a=torch.Tensor([1]).cuda()
b=torch.Tensor([1]).cuda()
c=[]
c.append(a)
c.append(b)

np.mean(c)

 

报错形式2：

 

使用GPU数据类型，需要在最后将列表转化为Tensor类型：