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