tensor和numpy、List基本转换

GTX_AI

tensor和numpy、List基本转换：
(1)cuda--cpu、tensor--numpy

1. pred_mask = (outputs*255).cpu().data.numpy().astype(np.uint8)

复制代码

(2)tensor大小和类型

1. print(outputs.type())
   
2. print(outputs.size())

复制代码

（3）numpy大小和类型

1. type(pred_mask)
   
2. print(pred_mask.shape)

复制代码