MemorySegmentTensorDataFactory

A TensorDataFactory that produces MemorySegmentTensorData tensors, keeping all data off-heap for SIMD-friendly access.

The factory manages a shared Arena whose lifetime should be tied to the owning execution context.

Constructors

constructor(arena: Arena = Arena.ofShared(), alignment: Long = 64)

open override fun close()

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

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

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

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

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

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

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

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

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

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