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