/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
python/torch_refs/torch_norm.py
24 строки
423 B
kolkir
Refactor python folder structure
01 мар 2025, 22:16
01 мар 2025, 22:16
ef25f05
Код
Авторство
О чём код?
import torch import torch.nn as nn import torch.nn.functional as F im = torch.tensor([[[[1, 2], [3, 4]], [[1, 2], [3, 4]]]], dtype=torch.float32) print(im.shape) norm = nn.BatchNorm2d(2) norm.train() out = norm(im) print("out:") print(out.shape) print(out) F.silu(out).mean().backward() print("grads:") print(norm.weight.grad) print(norm.bias.grad) print("runnings:") print(norm.running_mean) print(norm.running_var)