pytorch -- 张量的创建

pytorch框架中，有 4 种创建张量的方式

1.通过torch.tensor()函数创建

引入torch和numpy模块。

先创建了一个numpy数组a，紧接着创建了一个张量b，打印类型，结果如图。我们也可以进行类型转化，将numpy数组转化为tensor类型

用 to 可以在tensor类型间进行精度转化

 

2.用pytorch内置的函数创建张量

rand -- 服从 [0~1) 上均匀分布

randn -- 服从标准高斯分布

randint -- 按需分布

zeros -- 全0元素

ones -- 全1元素

eye -- 单位矩阵

 

3.通过已知张量创建出形状相同的张量

torch.rand_like(zeros_like、randn_like、ones_like...)

 

4.通过已知张量创建出形状不同但数据类型相同的张量

new_ones、new_randn、new_rand...