上传者: 38506182
|
上传时间: 2022-04-05 23:09:08
|
文件大小: 36KB
|
文件类型: pdf
1, 创建pytorch 的Tensor张量:
torch.rand((3,224,224)) #创建随机值的三维张量,大小为(3,224,224)
torch.Tensor([3,2]) #创建张量,[3,2]
2, cpu上的tensor和GPU即pytorch创建的tensor的相互转化
b = a.cpu() # GPU → CPU
a = b.cuda() #CPU → GPU
3, tensor和numpy的转化
b = a.numpy() # tensor转化为 numpy数组
a = b.from_numpy() # numpy数组转化为tensor
4, torch的