tensor主要有8个属性,如下图。4个与数据相关,4个与梯度求导相关。
.cuda表示是在gpu上。dtype表示张量的数据类型,一共9种,如下图
主要是3大类,float型,整型和布尔型。红框框出来的32位float和64位整型用的最多。32位float默认为权重的数据类型,64位整型用作标签的数据类型。
pytorch中张量的创建方法:
一、直接创建
其中pin_memory一般设置为False。
创建自定义数值张量(数值都相同):
等差数列的方法创建:
创建单位阵
依据概率分布创建:
创建正态分布:
创建标准正态分布:
创建均匀分布
创建0到n-1的随机排列,用于随机生成索引。还有就是生成伯努利分布
网友评论