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  23  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  1015  22
[torch.FloatTensor of size 2x2]
'''

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

pytorch 与 numpy 的相互转换相关推荐

pytorch和Numpy的区别以及相互转换
pytorch 数值转numpy int(x1.cpu().data.numpy()) pytorch和Numpy的区别以及相互转换 2018年01月17日 19:17:54 阅读数:2654
pytorch 与numpy 部分操作的对应关系
pytorch号称神经网络界的numpy,确实非常好用,容易上手,自己在学习中总结了 pytorch 与numpy 的一些运算操作上的不同. np.dot(a,b) ==> torch.mm( ...
Spearman’s correlation coefficient--斯皮尔曼相关系数pytorch与numpy实现
文章目录 Spearman's correlation介绍 Pytorch实现 Numpy实现 Spearman's correlation介绍斯皮尔曼等级相关(Spearman's correla ...
Pytorch和Numpy的默认类型
1.对于整形数据部分的默认类型区别 pytorch创建的整数tensor是torch.int64类型, 64位类型 Numpy创建的整数ndarrays是np.int32, 32位类型 2.对于浮 ...
用pytorch及numpy计算成对余弦相似性矩阵，并用numpy实现kmeans聚类
sklearn和scipy里面都提供了kmeans聚类的库,但是它们都是根据向量直接进行计算欧氏距离.闵氏距离或余弦相似度,如果使用其他的度量函数或者向量维度非常高需要先计算好度量距离然后再聚类时 ...
pytorch下Numpy，Torch，Spicy，NetworkX及其他基本数据类型相关操作(持续更新)
Tricks 1. torch.sparse.FloatTensor(position, value) 稀疏张量表示为一对稠密张量:一个值张量和一个二维指标张量(每一维中存储多个值).一个稀疏张量可以 ...
pytorch替换numpy中的一些组件 //转载请注明来源
numpy可以同时处理两种数据:list和numpy.ndarray. 而torch只能处理张量(Tensor)数据. 1.替换 np.asarray() import torch data=[[1, ...
pytorch和Numpy安装（不实用系列）
前言 !!!!!!!!!!!!!!!! 这篇文章其实很麻烦,当时我安装了许久,最近发现一个up主,十分靠谱!!!! !!!在b站有对应视频: 指路:https://www.bilibili.com/v ...
python实现与MATLAB接近的bicubic imresize（pytorch 与 numpy版）
An pytorch implementation of imresize function in MATLAB with bicubic kernel. pytorch 版本 import torc ...

pytorch 与 numpy 的相互转换

pytorch 与 numpy 的相互转换相关推荐

最新文章

热门文章