好资源导航 » 文章资讯 » pytorch 实现tensor与numpy数组转换

pytorch 实现tensor与numpy数组转换

2023-08-03 20:01:04 86

看代码，tensor转numpy:

a=torch.ones(2,2)
b=a.numpy()
c=np.array(a)#也可以转numpy数组
print(type(a))
print(type(b))
print(a)
print(b)

输出为：



tensor([[1.,1.],
[1.,1.]])
[[1.1.]
[1.1.]]

numpy转tensor：

importtorch
importnumpyasnp

a=np.ones(5)
b=torch.from_numpy(a)
c=torch.Tensor(a)#也可以转pytorchTensor
print(type(a))
print(type(b))
print(a)
print(b)

输出为：



[1.1.1.1.1.]
tensor([1.,1.,1.,1.,1.],dtype=torch.float64)

可见pytorch的tensor对象与numpy数组是可以相互转换的，且numpy数组的默认类型是double

以上这篇pytorch实现tensor与numpy数组转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。

声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。

返回顶部
514930285
czq8825@qq.com

pytorch 实现tensor与numpy数组转换

热门推荐

随机推荐