/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
with_cpu
python/torch_refs/torch_conv.py
28 строк
675 B
kolkir
Refactor python folder structure
01 мар 2025, 22:16
01 мар 2025, 22:16
ef25f05
Код
Авторство
О чём код?
import torch import torch.nn.functional as F im = torch.tensor([[[[1, 2, 3], [4, 5, 6], [7, 8, 9]]]], dtype=torch.float32) print(im.shape) kernel = torch.tensor([[7, 8], [1, 3]], dtype=torch.float32, requires_grad=True) kernel = kernel.view(1, 1, 2, 2) print(kernel.shape) bias = torch.ones([1], dtype=torch.float32, requires_grad=True) out = F.conv2d(im, kernel, bias) print(out) kernel.retain_grad() bias.retain_grad() out.mean().backward() print(kernel.grad) print(bias.grad) # kernel = torch.tensor([[[7, 8], [1, 3]], [[7, 8], [1, 3]]]) # kernel = kernel.view(2, 1, 2, 2) # bias = torch.ones([2], dtype=torch.long) # out = F.conv2d(im, kernel, bias) # print(out)