Pytorch 建立固定值Tensor的方法

2024-10-20 05:54:10

1、加载pytorch中的torch库,以便于使用其中的数据定义及相关函数使用指令【import torch】

Pytorch 建立固定值Tensor的方法

3、定义全0 tensor是使用zeros函数定义,使用指令【torch.zeros(dim)】

Pytorch 建立固定值Tensor的方法

5、建立全1tensor的方式和step3、step4类似,分别使用ones()和ones_like()使用指令【torch.ones(dims)】和【torch.ones_like(tensora)】

Pytorch 建立固定值Tensor的方法

7、建立eye tensor时,直接使用eye()建立即可。注意 eye只能建立方阵使用指令【torch.eye(dim)】

Pytorch 建立固定值Tensor的方法

8、以上,就是pytorch中建立一些具有确定数值的tensor的方法啦

猜你喜欢