SkainetScope

DSL Scope for Skainet that wraps a GraphExecutionContext and provides a more idiomatic way to interact with tensors and training loops.

Constructors

High-level recording block.

fun <T : DType, V> ~~tensorOf~~(shape: Shape, dtype: KClass<T>, vararg values: Float): Tensor<T, V>

Create a tensor with specific shape and values.

fun trainStep(optimizer: Optimizer, vararg params: ModuleParameter<*, *>, block: SkainetScope.() -> Tensor<*, *>): Tensor<*, *>

Encapsulated training step.

Mark a tensor as requiring gradients.