import torch
x = torch.randn(28, 28, 3)
layer = torch.nn.Conv2d(x, 32, kernel_size=3, stride=1, padding=((0,1),(0,1)))
out = layer(x)
It produces the error:
Traceback (most recent call last):
RuntimeError: bool value of Tensor with more than one value is ambiguous