torch和numpy的对比

一、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)

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容