ndarray
autograd
loss
Block.
forward
Overrides to implement forward computation using NDArray. Only accepts positional arguments.
NDArray
*args (list of NDArray) – Input tensors.