SKaiNET
Toggle table of contents
common
Target filter
common
Switch theme
Search in API
Skip to content
SKaiNET
skainet-compile-dag
/
sk.ainet.lang.graph.dsl
/
SkainetScope
/
trainStep
train
Step
fun
trainStep
(
optimizer
:
Optimizer
,
vararg
params
:
ModuleParameter
<
*
,
*
>
,
block
:
SkainetScope
.
(
)
->
Tensor
<
*
,
*
>
)
:
Tensor
<
*
,
*
>
(
source
)
Encapsulated training step.