skainet-lang-core/sk.ainet.java/TrainingLoop/step

step

fun step(x: Tensor<*, *>, y: Tensor<*, *>): Float(source)

Perform a single training step.

Return

The loss value as a float.

Parameters

x

Input tensor.

y

Target tensor.