美文网首页
torch和numpy的对比

torch和numpy的对比

作者: 平凡不普通 | 来源:发表于2020-10-27 07:46 被阅读0次

一、numpy和torch

numpy是python中处理数据的模块,可以处理各种的矩阵(matrix)。

Torch自称为神经网络中的numpy。它会将torch产生的tensor放在GPU中加速运算,就像numpy会把array放在CPU中加速运算。

二.numpy array与torch tensor之间的相互转换

array2tensor=torch.from_numpy(numpy_data)#numpy array->torch tensor,其参数必须是数组形式

tensor2array=torch_data.numpy()#torch tensor->numpy array

三. 用法对比

numpy 数据形式下矩阵a,位置变化用a.transpose(1,2,0)

tensor 数据格式下的矩阵b, 位置变化时 b.permute(1,2,0)

相关文章

网友评论

      本文标题:torch和numpy的对比

      本文链接:https://www.haomeiwen.com/subject/cqstvktx.html