skainet-lang-core/sk.ainet.lang.tensor.ops/TensorSpec

TensorSpec

data class TensorSpec(val name: String, val shape: List<Int>?, val dtype: String, val requiresGrad: Boolean = false, val metadata: Map<String, Any> = emptyMap())(source)

Specification for a tensor in the graph, containing shape and type information

Constructors

TensorSpec

constructor(name: String, shape: List<Int>?, dtype: String, requiresGrad: Boolean = false, metadata: Map<String, Any> = emptyMap())

Properties

dtype

val dtype: String

Data type of the tensor

metadata

val metadata: Map<String, Any>

Additional metadata for this tensor

name

val name: String

Name/identifier of this tensor

requiresGrad

val requiresGrad: Boolean

Whether this tensor requires gradients

shape

val shape: List<Int>?

Shape of the tensor (null for dynamic shapes)

tensorEncoding

val TensorSpec.tensorEncoding: TensorEncoding?

Physical storage encoding carried on this spec, or null if the producer did not populate it.

Functions

withTensorEncoding

fun TensorSpec.withTensorEncoding(encoding: TensorEncoding?): TensorSpec

Return a copy of this spec with encoding stored in its metadata map. Passing null removes the entry; passing a non-null value adds or replaces it, leaving all other metadata untouched.