与组卷积

Question

我正在使用tensorflow 1.13.1和python 3.7.1。如何将以下pytorch代码转换为tensorflow：

X = torch.ones(3,3,1024,1024) weights = torch.ones(3,1,3,3) Y = F.conv2d(X, weights, bias=torch.zeros(3), groups=3)