使用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
相關文章
相關標籤/搜索