skainet-lang-core/sk.ainet.lang.tensor.data/TensorDataFactory

TensorDataFactory

interface TensorDataFactory(source)

Interface for tensor data factory used by the DSL

Inheritors

DenseTensorDataFactory

MemorySegmentTensorDataFactory

Functions

fromByteArray

abstract fun <T : DType, V> fromByteArray(shape: Shape, dtype: KClass<T>, data: ByteArray): TensorData<T, V>

fromFloatArray

abstract fun <T : DType, V> fromFloatArray(shape: Shape, dtype: KClass<T>, data: FloatArray): TensorData<T, V>

fromIntArray

abstract fun <T : DType, V> fromIntArray(shape: Shape, dtype: KClass<T>, data: IntArray): TensorData<T, V>

full

abstract fun <T : DType, V> full(shape: Shape, dtype: KClass<T>, value: Number): TensorData<T, V>

init

abstract fun <T : DType, V> init(shape: Shape, dtype: KClass<T>, generator: (indices: IntArray) -> V): TensorData<T, V>

ones

abstract fun <T : DType, V> ones(shape: Shape, dtype: KClass<T>): TensorData<T, V>

randn

abstract fun <T : DType, V> randn(shape: Shape, dtype: KClass<T>, mean: Float, std: Float, random: Random): TensorData<T, V>

randomInit

abstract fun <T : DType, V> randomInit(shape: Shape, dtype: KClass<T>, generator: (random: Random) -> V, random: Random): TensorData<T, V>

uniform

abstract fun <T : DType, V> uniform(shape: Shape, dtype: KClass<T>, min: Float, max: Float, random: Random): TensorData<T, V>

wrapByteArray

open fun <T : DType, V> wrapByteArray(shape: Shape, dtype: KClass<T>, data: ByteArray): TensorData<T, V>

Wraps a ByteArray without copying. The caller must ensure the array is not mutated while the returned TensorData is in use. Default implementation falls back to fromByteArray (which copies).

wrapFloatArray

open fun <T : DType, V> wrapFloatArray(shape: Shape, dtype: KClass<T>, data: FloatArray): TensorData<T, V>

Wraps a FloatArray without copying. The caller must ensure the array is not mutated while the returned TensorData is in use. Default implementation falls back to fromFloatArray (which copies).

wrapIntArray

open fun <T : DType, V> wrapIntArray(shape: Shape, dtype: KClass<T>, data: IntArray): TensorData<T, V>

Wraps an IntArray without copying. The caller must ensure the array is not mutated while the returned TensorData is in use. Default implementation falls back to fromIntArray (which copies).

zeros

abstract fun <T : DType, V> zeros(shape: Shape, dtype: KClass<T>): TensorData<T, V>