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

Pytorch学习笔记(二)——创建Tensor

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

Pytorch学习笔记(二)——创建Tensor

学习目标:

掌握Pytorch


学习内容:

Tensor的创建方式有四种,分别是torch.Tensor()、torch.tensor()、torch.as_tensor()、torch.from_numpy()。
1、torch.Tensor()
torch.Tensor()为类构造函数,创建后,默认dtype为float32。

data = np.ones([2,2],dtype = np.int64)
# [[1 1]
#  [1 1]]

#采用torch.Tensor()创建一个tensor
torch.Tensor(data)
# tensor([[1., 1.],
#         [1., 1.]])

2、torch.tensor()
torch.tensor()为工厂函数。工厂函数可以显式的设置数据类型,但是构造类函数不能直接设置。

#采用torch.tensor()创建一个tensor
torch.tensor(data,dtype = torch.float16)
# tensor([[1., 1.],
#        [1., 1.]], dtype=torch.float16)

3、torch.as_tensor()
torch.as_tensor()为工厂函数。

#采用torch.tensor()创建一个tensor
torch.as_tensor(data)
# tensor([[1, 1],
#         [1, 1]])

4、torch.from_numpy()
torch.from_numpy()为工厂函数。

#采用orch.from_numpy()创建一个tensor
torch.from_numpy(data)
# tensor([[1, 1],
#         [1, 1]])

Note:类构造函数与工厂函数的区别在于数据类型不同,工厂函数会根据输入推断数据类型。


总结:

要坚持要加油呀~

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

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

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