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

Package-level declarations

Types

ConvKernel

class ConvKernel<T : DType, V>(builder: TensorBuilder<T, V>)

Creates convolution kernel tensors

DataContextDsl

interface DataContextDsl : ContextDslItem

DataDefinitionContextDslImpl

class DataDefinitionContextDslImpl(val executionContext: ExecutionContext) : DataContextDsl

ImageInitializers

class ImageInitializers<T : DType, V>(shapeBuilder: ShapeBuilder<T, V>)

Common image processing initializations

ImageTensor

class ImageTensor<T : DType, V>(builder: TensorBuilder<T, V>)

Creates image tensors in BCHW (Batch, Channel, Height, Width) format

InitializationType

sealed class InitializationType<out V>

Types of tensor initialization

MLPBias

class MLPBias<T : DType, V>(builder: TensorBuilder<T, V>)

Creates bias tensors for MLP layers

MLPWeights

class MLPWeights<T : DType, V>(builder: TensorBuilder<T, V>)

Creates weight tensors for MLP layers with common initialization strategies

ShapeBuilder

class ShapeBuilder<T : DType, V>(dtype: KClass<T>, shape: Shape)

Builder for tensors with a defined shape, allowing initialization

TensorBuilder

class TensorBuilder<T : DType, V>(dtype: KClass<T>)

Builder class for constructing tensors with various initialization strategies

TensorContextDslItem

interface TensorContextDslItem

TensorCreationScope

interface TensorCreationScope<T : DType, V>

TensorCreationScopeImpl

class TensorCreationScopeImpl<T : DType, V>(val executionContext: ExecutionContext, val shape: Shape, val dtype: KClass<T>) : TensorCreationScope<T, V>

Implementation of TensorCreationScope

TensorDefineDsl

interface TensorDefineDsl<T : DType, V> : TensorContextDslItem

TensorDsl

@DslMarker

annotation class TensorDsl

TensorFactoryContext

class TensorFactoryContext<T : DType, V>(executionContext: ExecutionContext, dtype: KClass<T>)

Context wrapper that provides tensor data factory access

TensorInitializer

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

Initializer that holds the complete tensor specification

TypedDataContextDsl

interface TypedDataContextDsl<T : DType, V> : DataContextDsl

TypedDataContextDslImpl

class TypedDataContextDslImpl<T : DType, V>(baseExecutionContext: ExecutionContext, val defaultDType: KClass<T>) : TypedDataContextDsl<T, V>

Functions

bias

fun <T : DType, V> TensorBuilder<T, V>.bias(): MLPBias<T, V>

Extension function to access MLP bias creation

image

fun <T : DType, V> TensorBuilder<T, V>.image(): ImageTensor<T, V>

Extension function to access image tensor creation

imageInit

fun <T : DType, V> ShapeBuilder<T, V>.imageInit(): ImageInitializers<T, V>

Extension function to access image-specific initializers

kernel

fun <T : DType, V> TensorBuilder<T, V>.kernel(): ConvKernel<T, V>

Extension function to access convolution kernel creation

matrix

inline fun <T : DType, V> DataContextDsl.matrix(rows: Long, columns: Long, noinline content: TensorCreationScope<T, V>.() -> Tensor<T, V>): Tensor<T, V>

scalar

inline fun <T : DType, V> DataContextDsl.scalar(value: V): Tensor<T, V>

tensor

fun <T : DType, V> tensor(executionContext: ExecutionContext, dtype: KClass<T>, content: TensorDefineDsl<T, V>.() -> Tensor<T, V>): Tensor<T, V>

tensor

inline fun <T : DType, V> DataContextDsl.tensor(noinline content: TensorFactoryContext<T, V>.() -> Tensor<T, V>): Tensor<T, V>

inline fun <T : DType, V> DataContextDsl.tensor(name: String, noinline content: TensorFactoryContext<T, V>.() -> Tensor<T, V>): Tensor<T, V>

vector

inline fun <T : DType, V> DataContextDsl.vector(length: Long, noinline content: TensorCreationScope<T, V>.() -> Tensor<T, V>): Tensor<T, V>

weights

fun <T : DType, V> TensorBuilder<T, V>.weights(): MLPWeights<T, V>

Extension function to access MLP weight creation