pytorch 张量

2020-02-12  本文已影响0人  米线织毛衣

tensor主要有8个属性,如下图。4个与数据相关,4个与梯度求导相关。

.cuda表示是在gpu上。dtype表示张量的数据类型,一共9种,如下图

主要是3大类,float型,整型和布尔型。红框框出来的32位float和64位整型用的最多。32位float默认为权重的数据类型,64位整型用作标签的数据类型。

pytorch中张量的创建方法:

一、直接创建

其中pin_memory一般设置为False。

创建自定义数值张量(数值都相同):

等差数列的方法创建:

创建单位阵

依据概率分布创建:

创建正态分布:

创建标准正态分布:

创建均匀分布

创建0到n-1的随机排列,用于随机生成索引。还有就是生成伯努利分布

上一篇下一篇

猜你喜欢

热点阅读