您好,欢迎来到华拓科技网。
搜索
您的当前位置:首页基础语法(pytorch的tensor与Numpy)

基础语法(pytorch的tensor与Numpy)

来源:华拓科技网

tensor : 张量

1.Pytorch

Pytorch是由研发的开源的深度学习框架,并且支持CPU和GPU加速计算

Pytorch安装参考:get-started、locally

2.torch.Tensor

torch.tensor是包含单一数据类型元素的数组,它是Pytorch中数据存储和运算的基本单元,

类似于Numpy的numpy.ndarray,两者的函数接口和运算方式几乎一样,实际上Pytorch内部也是把ndarray包装成Tensor

3.Tensor

中文称为张量,(可以看做一个数组),是Pytorch中最基本的数据类型,标量,向量和矩阵都可以看做是张量的特例

torch定义了10种具有CUP和GPU类型的张量类型

import torch
import numpy as np
a = np.array([[1,2],[3,4]])
# //定义numpy的二维数组
print(type(a))
print(a)
print()
b = torch.tensor([[1,2],[3,4]])
#//定义二维torch的张量
print(type(b))
print(b)
print()

<class 'numpy.ndarray'>
[[1 2]
 [3 4]]

<class 'torch.Tensor'>
tensor([[1, 2],
        [3, 4]])

c = b.numpy()       #张量tensor转numpy
print(type(b))
print(b)
print()
d = torch.from_numpy(c)   #numpy数组转tensor张量
print(type(d))
print(d)
print()

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- huatuo6.cn 版权所有 赣ICP备2024042791号-9

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务