比较Pytorch和我的卷积实现
Comparison of my convolution implementation to pytorch
the Test code for pytorch conv2d
# pytorch
import torch
import torch.nn as nn
x = torch.tensor(range(50),dtype=torch.float)
x = torch.reshape(x, (1,2,5,5)) #N,C,H,W
x.requires_grad = True
print(x.s...