1.2 pytorch的入门使用
Pytorch的入门使用
目标
知道张量和Pytorch中的张量知道pytorch中如何创建张量知道pytorch中tensor的常见方法知道pytorch中tensor的数据类型知道pytorch中如何实现tensor在cpu和cuda中转化
1. 张量Tensor
张量是一个统称,其中包含很多类型:
0阶张量:标量、常数,0-D Tensor1阶张量:向量,1-D Tensor2阶张量:矩阵,2-D Tensor3阶张量…N阶张量
2. Pytorch中创建张量
使用python中的列表或者序列创建tensor torch.tensor([[1., -1.], [1., -1.]])
tensor([[ 1.0000, -1.0000],
[ 1.0000, -1.0000]])
使用numpy中的数组创建tensor torch.tensor(np.array([[1,
共有 0 条评论