Как заполнить диагональ значением в факеле? В numpy вы можете сделать:
a = np.zeros((3, 3), int)
np.fill_diagonal(a, 5)
array([[5, 0, 0],
[0, 5, 0],
[0, 0, 5]])
Я знаю, что torch.diag()
возвращает диагональ, но как использовать это как маску для присвоения новых значений, мне не под силу. Я не смог найти ответ здесь или в документации PyTorch.