torch:amax

(torch:amax a &key axis keepdims)

Differentiable maximum, of every element or along an axis (linalg:amax's rules). The gradient flows to every element equal to the maximum, split evenly among ties (PyTorch's amax rule).