skainet-lang-core/sk.ainet.lang.tensor.dsl/TensorInitializer

TensorInitializer

class TensorInitializer<T : DType, V>(dtype: KClass<T>, shape: Shape, initType: InitializationType<V>)(source)

Initializer that holds the complete tensor specification

Constructors

TensorInitializer

constructor(dtype: KClass<T>, shape: Shape, initType: InitializationType<V>)

Functions

fun build(executionContext: ExecutionContext): Tensor<T, V>

Build the actual tensor with the specified initialization