Package org.jetbrains.kotlinx.dl.api.core.dsl

Types

GraphTrainableModelBuilder

class GraphTrainableModelBuilder<T : GraphTrainableModel>(createModelFunction: (Array<Layer>) -> T)

Graph trainable model builder.

LayerListBuilder

value class LayerListBuilder(layers: MutableList<Layer>)

Layer list builder.

Functions

functional

fun functional(builder: GraphTrainableModelBuilder<Functional>.() -> Unit): Functional

Functional model builder.

invoke

operator fun <T : GraphTrainableModel> (Array<Layer>) -> T.invoke(builder: GraphTrainableModelBuilder<T>.() -> Unit): T

Generic model builder.

sequential

fun sequential(builder: GraphTrainableModelBuilder<Sequential>.() -> Unit): Sequential

Sequential model builder.