How do we figure out if backpropagation can work with such a layer, if we insert it into our network? In this particular, case, the documentation for torch.squeeze() doesn’t say whether PyTorch knows how to compute gradients for this operation. Is there a master list of PyTorch functions for which PyTorch knows how to compute gradients?
More generally, how can we ascertain if a custom nn.Module plays well with autograd?