捡历史的小木板

关注

PyTorch | torch和numpy之间的转换

捡历史的小木板

关注

阅读 71

2022-01-31

目录

1 tensor->array(tensor.numpy())

2 array->tensor(torch.from_numpy(array))


1 tensor->array(tensor.numpy())

x=torch.ones(3,2)
y=x.numpy()
print(x)
print(y)

底层是一样的数据

x.add_(1)
print(x)
print(y)

但是,如果不用add命令,而是用+,则两者又会不一样

x=x+z
print(x)
print(y)

2 array->tensor(torch.from_numpy(array))

m=np.ones((3,2))
n=torch.from_numpy(m)
print(m)
print(n)

相关推荐

一世独秀

PyTorch数据和NumPy数据之间的互操作

一世独秀 98 0 0

扬帆远航_df7c

莫烦pytorch学习笔记(一)——torch or numpy

扬帆远航_df7c 43 0 0

Xin_So

pytorch张量和numpy数组相互转换

Xin_So 11 0 0

梯梯笔记

pytorch Tensor在cpu和gpu之间转换

梯梯笔记 69 0 0

登高且赋

每天五分钟深度学习pytorch:tensor与python、numpy之间的转换

登高且赋 7 0 0

得一道人

pandas的Datafram与numpy数组之间如何转换

得一道人 63 0 0

小a草

numpy和torch读彩色图像区别

小a草 75 0 0

mjjackey

【pytorch 中 torch.max 和 torch.argmax 的区别】

mjjackey 19 0 0

早安地球

numpy的repeat和pytorch的repeat

早安地球 80 0 0

程序员知识圈

pytorch中将tensor转换为numpy时报错

程序员知识圈 197 0 0

精彩评论(0)

0 0 举报