tensor
open override fun tensorTrainingStep(tensor: DTensor, gradient: DTensor): DTensor
Content copied to clipboard
Train an element of the model. For the model itself, use train.
Train an element of the model. For the model itself, use train.