每天五分钟深度学习框架pytorch：如何创建多维Tensor张量元素？

作者：繁依Fanyi0 | 2024-05-07 03:58:55

踩

本文重点

上节课程我们学习了如何创建Tensor标量，我们使用torch.tensor。本节课程我们学习如何创建Tensor向量，我们即可以使用torch.Tensor又可以使用torch.tensor，下面我们看一下二者的共同点和不同点。

tensor张量是一个多维数组，零维就是一个点（就是上一章的标量），一维就是向量，二维就是一般的矩阵，多维就相当于一个多维的数组，这和 numpy理解是一样的，不同的是Tensor不仅可以在CPU上跑，在GPU上也可以跑。

方式一：将numpy转成tensor


import torch
import numpy as np
a=np.zeros((1,4))
b=torch.from_numpy(a)
print(b.dim())
print(b.shape)
print(b.size())

方式二：将列表，元组，数据转成tensor，常用于两维以下

impor

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/547302