padding = k2.shape[-1] - 1 # Flip because this is actually correlation, and permute to adapt to BHCW k3 = torch.conv2d(k1.permute(1, 0, 2, ...
確定! 回上一頁