メモ: PyTorch tensor requires_gradのTrue/False確認、切り替え

requires_gradのTrue/False確認

属性 requires_gradを参照して確認できる。

import torch

a = torch.tensor([0, 1], dtype=torch.float32)
b = torch.tensor([2, 3], dtype=torch.float32, requires_grad=True)

a.requires_grad
b.requires_grad

実行結果

False
True

requires_gradのTrue/False切り替え

メソッド requires_grad()で切り替えられる

a.requires_grad_(True)
a.requires_grad

True # <- FalseからTrueに変わった