Skip to main content

//clika-runtime/io.clika.runtime/Ops/amax

amax

[common]
fun amax(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false): Tensor

amax(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false): the amax operator. Maximum value of input over dims. Empty dims reduces every dimension. Returns VALUES (for the positions use argmax).