栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch张量创建:区分Tensor & tensor & as

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch张量创建:区分Tensor & tensor & as

本文内容分为以下两个部分:
代码验证
结论

代码验证

请看如下代码:

import torch
import numpy as np

data = np.array([1,2,3])
t1 = torch.Tensor(data)
t2 = torch.tensor(data)
t3 = torch.as_tensor(data)
t4 = torch.from_numpy(data)

print(t1)
print(t2)
print(t3)
print(t4)

输出的结果是:

tensor([1.,2.,3.])
tensor([1,2,3],dtype=torch.int32)
tensor([1,2,3],dtype=torch.int32)
tensor([1,2,3],dtype=torch.int32)

再输入以下语句查看类型:

print(t1.dtype)
print(t2.dtype)
print(t3.dtype)
print(t4.dtype)

输出的结果是:

torch.float32
torch.int32
torch.int32
torch.int32

接着,修改numpy数组data的值,输入代码如下:

data[0] = 0
data[1] = 0
data[2] = 0

print(t1)
print(t2)
print(t3)
print(t4)

输出后发现仅有t3,t4的值被改变了,说明as_tensor和from_numpy的张量是和numpy数组共用的存储空间,也可以称为“share data”。

结论
特性Tensortensoras_tensorfrom_numpy
是否与numpy共用存储
是否属于factory function
深度学习适用场景/写代码常用优化性能使用优化性能使用

注:
(1) factory function是“先判断输入数据类型,后按照相应类型存储”的函数,而Tensor是一个构造函数;
(2) 优化程序性能时通常选用as_tensor而不是from_numpy,因为后者只接受numpy数组。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/853319.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号