ndarray
同理于Numpy的ndarray
,在Pytorch中被称为是torch
,其提供了数学上的张量的概念。
- x = torch.arrange(n) : 获取1~n的一维(1*n)张量
- x = torch.zeros/ones/randn(x,y,z) :创建(xyz)大小的全零,全一,随机张量
- x = torch.tensor([1,2,3],[4,5,6]) : 自定义初始化一个张量
- x.shape
- x.numel() 获取元素总数量
- x.reshape(r,c)
- x[1:2,:] :切片,使用切片时可以节省内存