torch:amax

(torch:amax a &key axis keepdims)

微分可能な最大値です。全要素、または軸に沿って計算します (linalg:amax の規則)。勾配は最大値に等しいすべての要素に流れ、同値の間では均等に分配されます (PyTorch の amax の規則)。