pytorch中torch.FloatTensor的用法

  1. Pytorch中只能对Variable类型变量求导
    pytorch中torch.FloatTensor的用法_第1张图片
    对x进行类型转换后可进行求导
    pytorch中torch.FloatTensor的用法_第2张图片
    但是直接求导还不行,因为x是一个矩阵。所以应该在backward函数里加上torch.FloatTensor,并指定求导后,梯度伸缩倍数。

注意,求梯度只针对于浮点型tensor,整型tensor不能求梯度
requires_grad只针对浮点型tensor

你可能感兴趣的:(pytorch)