pytorch中.weight.data与.weight的区别

pytorch中.weight.data与.weight

.weight.data:得到的是一个Tensor的张量(向量),不可训练的类型
.weight:得到的是一个parameter的变量,可以计算梯度的

  • 查看两个方法的类型
    在这里插入图片描述
    在这里插入图片描述

  • 查看两个方法的值
    在这里插入图片描述
    pytorch中.weight.data与.weight的区别_第1张图片

你可能感兴趣的:(pytorch,深度学习)