1. 首页
  2. 数据库
  3. 其它
  4. PyTorch中张量的创建方法

PyTorch中张量的创建方法

上传者: 2020-12-23 02:06:16上传 PDF文件 77.69KB 热度 14次
张量的概念 张量,即Tensor,是PyTorch的基本数据结构。在数学概念中,张量是一个多维数组,它是标量、向量、矩阵的高维拓展。 torch.Tensor中的属性: data: 被包装的Tensor grad: data的梯度 grad_fn: 创建Tensor的Function,如加法,乘法,这个操作在求导过程中需要用到,所以需要将其记录下来。 requires_grad: 指示是否需要计算梯度 is_leaf: 指示是否是叶子结点 dtype: 张量的数据类型,如 torch.FloatTensor, torch.cuda.FloatTensor shape: 张量的形状,如(64
用户评论