![](https://img.haomeiwen.com/i11963149/adbd057a11dcd8aa.png)
tensor主要有8个属性,如下图。4个与数据相关,4个与梯度求导相关。
![](https://img.haomeiwen.com/i11963149/6a1b5076bcb803c3.png)
.cuda表示是在gpu上。dtype表示张量的数据类型,一共9种,如下图
![](https://img.haomeiwen.com/i11963149/ffcd62f220035144.png)
主要是3大类,float型,整型和布尔型。红框框出来的32位float和64位整型用的最多。32位float默认为权重的数据类型,64位整型用作标签的数据类型。
pytorch中张量的创建方法:
![](https://img.haomeiwen.com/i11963149/a09fce61b1c68f44.png)
一、直接创建
![](https://img.haomeiwen.com/i11963149/d131bcb17cf1c9b0.png)
其中pin_memory一般设置为False。
![](https://img.haomeiwen.com/i11963149/ee2a8aa92fc7070a.png)
![](https://img.haomeiwen.com/i11963149/1f5f8874bc9aef95.png)
![](https://img.haomeiwen.com/i11963149/b4dfd67bee6dd24c.png)
![](https://img.haomeiwen.com/i11963149/5e33228dfe856972.png)
创建自定义数值张量(数值都相同):
![](https://img.haomeiwen.com/i11963149/4e5c9b439576476b.png)
等差数列的方法创建:
![](https://img.haomeiwen.com/i11963149/7fb97919bdee5e88.png)
![](https://img.haomeiwen.com/i11963149/5b4039bf279e86d2.png)
![](https://img.haomeiwen.com/i11963149/5f6f1e6fa4275743.png)
创建单位阵
![](https://img.haomeiwen.com/i11963149/aec8439f19140661.png)
依据概率分布创建:
创建正态分布:
![](https://img.haomeiwen.com/i11963149/0ed52f7c3097e6fd.png)
![](https://img.haomeiwen.com/i11963149/84cee49dba21d787.png)
创建标准正态分布:
![](https://img.haomeiwen.com/i11963149/bed479a6c09a1da3.png)
创建均匀分布
![](https://img.haomeiwen.com/i11963149/108494d1807f6fdb.png)
创建0到n-1的随机排列,用于随机生成索引。还有就是生成伯努利分布
![](https://img.haomeiwen.com/i11963149/01318934d5cbecee.png)
网友评论