PyTorch Forums
Confused about torch.max() and gradient
autograd
fiona_H
(Fiona.H)
February 25, 2019, 4:47pm
4
z,y = torch.max(x,1)
So that is the reason
y
doesn’t have a gradient function?
show post in topic