使用torch.Tensor() 建立张量,加上requires_grad参数报错(Pytorch 1.0)

x = torch.Tensor([[.5, .3, 2.1]]) print(x) > tensor([[0.5000, 0.3000, 2.1000]]) 加上参数 requires_grad=True 或者 requires_grad=False :python x = torch.Tensor([[.5, .3, 2.1]], requires_grad=False) print(x) T
相关文章
相关标签/搜索