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,:] :切片,使用切片时可以节省内存