pytorch 与 numpy 的相互转换

# -*- coding: utf-8 -*- # @Time    : 2018/1/17 16:37 # @Author  : Zhiwei Zhong # @Site    : # @File    : Numpy_Pytorch.py # @Software: PyCharm import torch import numpy as np np_data = np.arange(6).reshape((2, 3)) # numpy 转为 pytorch格式 torch_data = torch.from_numpy(np_data) print( '\n numpy', np_data, '\n torch', torch_data, ) '''
 numpy [[0 1 2]
 [3 4 5]] 
 torch 
 0  1  2
 3  4  5
[torch.LongTensor of size 2x3]
''' # torch 转为numpy tensor2array = torch_data.numpy() print(tensor2array) """
[[0 1 2]
 [3 4 5]]
""" # 运算符 # abs 、 add 、和numpy类似 data = [[1, 2], [3, 4]] tensor = torch.FloatTensor(data) # 转为32位浮点数,torch接受的都是Tensor的形式,所以运算前先转化为Tensor print( '\n numpy', np.matmul(data, data), '\n torch', torch.mm(tensor, tensor) # torch.dot()是点乘 ) '''
 numpy [[ 7 10]
 [15 22]] 
 torch 
  7  10
 15  22
[torch.FloatTensor of size 2x2]
'''

 转自:https://blog.csdn.net/qq_34535410/article/details/79088952

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值