public argmax() as Int:
Returns the flat index of the maximum element.
Zero-based row-major index.
import klyn.math a = NDArray<Int>.fromList([[7, 3], [4, 9]]) print(a.argmax()) # 3