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

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

Properties

val dtype: String

Data type of the tensor

val metadata: Map<String, Any>

Additional metadata for this tensor

val name: String

Name/identifier of this tensor

val requiresGrad: Boolean

Whether this tensor requires gradients

val shape: List<Int>?

Shape of the tensor (null for dynamic shapes)