/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
python/torch_refs/torch_grad_unary.py
31 строка
524 B
kolkir
Refactor python folder structure
01 мар 2025, 22:16
01 мар 2025, 22:16
ef25f05
Код
Авторство
О чём код?
import torch def log_softmax(x): return torch.log_softmax(x, dim=2) def show_grad(x_val, op): x = torch.tensor(x_val, requires_grad=True) print(f"x shape {x.shape}, x={x}") z = op(x) print(f"z={z}") z.mean().backward() print(f"x grad = {x.grad}") print("log_softmax") show_grad( [ [ [1.0, 2.0, 3.0, 4.0], [5.0, 6.0, 7.0, 8.0], ], [ [1.0, 2.0, 3.0, 4.0], [5.0, 6.0, 7.0, 8.0], ], ], log_softmax, )