torch和numpy的对比
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch和numpy的对比相关的知识，希望对你有一定的参考价值。
参考技术A 一、numpy和torchnumpy是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的对比的主要内容，如果未能解决你的问题，请参考以下文章
python中choice对比（在numpy和random中均出现过）外加sample()函数对比
深度 初学者怎么选择神经网络环境？对比MATLABTorch和TensorFlow