Tensor

Tensor称为神经网络界的Numpy, 它与Numpy相似，二者共享内存，互相转换非常方便和高效
与numpy
下载

  1. pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

好.png
缺点:没有Div

创建tensor(张量)

注意小写的tensor只接受现有的数据；而大写的Tensor相当于就是FloatTensor，既可以接收现有的数据，也可以接受shape来创建指定形状的Tensor。

使用numpy创建

torch.tensor([],dtype=,device

  1. import torch
  2. import numpy as np
  3. #利用numpy创建
  4. a = np.array([1, 2])
  5. a_torch = torch.from_numpy(a)
  6. print(type(a), type(a_torch))

torch的组成

  • torch 类似于Numpy的通用数组库,可以在GPU上运行
  • torch.autograd 自动求取梯度
  • torch.nn 具有共享层和损失函数的神经网络库
  • torch.optim 具有通用优化算法的优化包

    Tensor

Tensor称为神经网络界的Numpy, 它与Numpy相似，二者共享内存，互相转换非常方便和高效

1 创建tensor

  1. import torch
  2. torch.tensor(data, dtype=None, device=None, requires_grad=False)
  3. torch.tensor # 会从data中的数据部分做拷贝（而不是直接引用），根据原始数据类型生成相应的torch.LongTensor、torch.FloatTensor和torch.DoubleTensor