PyTorch的安装、CUDA环境的配置以及Tensor的基本操作请查看「深度学习」PyTorch笔记-01-基础知识。Tensor基本操作创建直接输入值创建Tensor:import torch
torch.Tensor([[1, 2], [3, 4]])
Out[3]:
tensor([[1., 2.],
[3., 4.]])打印Tensor类型:a = torch.Tensor([[1, 2], [3, 4]])
a.type(), type(a)
Out[5]: ('torch.FloatTensor', torch.Tensor)创建全为1的Tensor:torch.ones(2, 3)
Out[11]:
tensor([[1., 1., 1.],
[1., 1., 1.]])创建一个与指定Tensor形状相同且全1的Tensor:torch.ones_like(a)
Out[12]:
tensor([[1., 1.],
[1., 1.]])创建全为0的Tensor:torch.zeros(3, 2)
Out[13]:
tensor([[0., 0.],
[0., 0.],
[0., 0.]])创建一个与指定Tensor形状相同且全0的Tensor:torch.zeros_l
...
继续阅读
(46)