pytorch课程6:基本数据类型

pytorch课程6:基本数据类型,第1张

pytorch课程6:基本数据类型  一、数据类型

 pytorch没有自带的string的类型,需要用以下二种编码方法表示:

 基本数据类型

 红框的三个最常用,其中ByteTensor 用来比较输入的两个值,如果相等返回1,否则返回0.

查看和检验某个变量的数据类型

 

torch 默认的数据类型是floattensor

修改默认设置用:torch.set_default_tensor_type(torch.DoubleTensor)

增强学习中一般用double,其他一般用float。

 二、标量(dimension为0)

 

 求一个标量的长度还可以用 a.dim()

三、向量 1、一维向量 dim=1

 

 2、二维向量 dim=2

   randn()为随机正态分布

  3、三维向量 dim=3

  4、四维向量 dim=4

 二、创建tensor 1.通过np的数组创建:torch.from_numpy( )

 2.通过列表创建

  

 

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5671129.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存