convolution operation pytorch code example
Example: convolution operation pytorch
import torch.nn.functional as F
from torch.autograd import Variable
F.conv2d(Variable(a.view(1,1,3,3)),Variable(b.view(1,1,3,3)))
import torch.nn.functional as F
from torch.autograd import Variable
F.conv2d(Variable(a.view(1,1,3,3)),Variable(b.view(1,1,3,3)))