skainet-io-safetensors/sk.ainet.io.safetensors/ShardedTensorInfo

ShardedTensorInfo

data class ShardedTensorInfo(base: StreamingSafeTensorInfo, val shardFilename: String, val shardIndex: Int, val totalShards: Int)(source)

Tensor info with shard location information.

Wraps StreamingSafeTensorInfo and adds information about which shard contains this tensor.

Constructors

ShardedTensorInfo

constructor(base: StreamingSafeTensorInfo, shardFilename: String, shardIndex: Int, totalShards: Int)

Properties

absoluteDataOffset

val absoluteDataOffset: Long

val dataOffsetEnd: Long

dataOffsetStart

val dataOffsetStart: Long

val dataType: DataType

val dtype: String

val elementCount: Long

val isUnknownType: Boolean

val name: String

val shape: List<Long>

val shardFilename: String

Filename of the shard containing this tensor

val shardIndex: Int

1-based shard index

val shardLocation: String

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

val sizeInBytes: Int

val totalShards: Int

Total number of shards