ndarray
autograd
loss
NDArray.
argmax
Convenience fluent method for argmax().
argmax()
The arguments are the same as for argmax(), with this array as data.