soulitzer:

import torch
a = torch.rand(10, 5, 5, 5, requires_grad=True)
b = torch.rand(5, 5, 2, 2, requires_grad=False)
out = torch.max(torch.nn.functional.conv2d(a, b)) 
out.requires_grad # True



Thank you for the sample! I went back to check line by line and found it’s another intermediate operation that does not support backpropagation and have it fixed now.