ndarray
autograd
loss
Parameter.
cast
Cast data and gradient of this Parameter to a new data type.
dtype (str or numpy.dtype) – The new data type.