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...