skainet-io-core/sk.ainet.io.model/TensorInfo

TensorInfo

data class TensorInfo(val name: String, val shape: List<Long>, val dataType: DataType, val elementCount: Long, val sizeInBytes: Long?, val format: ModelFormat, val nativeDType: String? = null, val skainetDType: String? = null, val canLoadNatively: Boolean = false, val sourceFile: String? = null, val shardIndex: Int? = null, val totalShards: Int? = null)(source)

Unified tensor information across all model formats.

This class provides a format-agnostic representation of tensor metadata, enabling consistent handling regardless of the source format (ONNX, GGUF, etc.).

Constructors

constructor(name: String, shape: List<Long>, dataType: DataType, elementCount: Long, sizeInBytes: Long?, format: ModelFormat, nativeDType: String? = null, skainetDType: String? = null, canLoadNatively: Boolean = false, sourceFile: String? = null, shardIndex: Int? = null, totalShards: Int? = null)

Properties

canLoadNatively

val canLoadNatively: Boolean

True if this tensor can be loaded into SKaiNET without conversion

val dataType: DataType

The unified DataType for display and processing

val elementCount: Long

Total number of elements in the tensor

val format: ModelFormat

The source model format

isFromShardedModel

val isFromShardedModel: Boolean

Whether this tensor is from a sharded model

val name: String

The tensor name

val nativeDType: String?

The original dtype name from the source format (e.g., "F32", "FLOAT")

val shape: List<Long>

The tensor dimensions

val shapeString: String

Human-readable shape string, e.g., "3, 224, 224"

val shardIndex: Int?

For sharded models, the 1-based shard index

val shardLocation: String?

Formatted shard location, e.g., "1/3"

val sizeInBytes: Long?

Size in bytes (null if unknown or variable)

val sizeString: String

Human-readable size string, e.g., "1.5 MB"

val skainetDType: String?

The corresponding SKaiNET DType name, or null if not supported

val sourceFile: String?

For sharded models, the shard filename containing this tensor

val totalShards: Int?

For sharded models, the total number of shards