pt

獲取梯度 x = torch.ones(2, 2, requires_grad=True) y = x + 2 z = y * y * 3 out = z.mean() out.backward() print(x.grad) 最底層的x需要grad，所以它的上層自然默認求梯度 取出tensor x = torch.randn(1) print(x.item()) 只有一個元素用item取出數字

>>阅读原文<<