I believe this thread answers your question (max is an identity function for the maximum element).
Confused about torch.max() and gradient
1 Like
I believe this thread answers your question (max is an identity function for the maximum element).
Confused about torch.max() and gradient