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