SKaiNET
Toggle table of contents
common
Target filter
common
Switch theme
Search in API
Skip to content
SKaiNET
skainet-lang-core
/
sk.ainet.lang.nn.dsl
/
TrainingRunner
/
step
step
fun
step
(
ctx
:
ExecutionContext
,
x
:
Tensor
<
T
,
V
>
,
y
:
Tensor
<
out
DType
,
*
>
)
:
Tensor
<
T
,
V
>
(
source
)
Perform a single training step: forward, backward, optimizer step, and zero grad.