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

TensorInfo

data class TensorInfo(val name: String, val shape: Shape, val dtype: String, val offset: Long, val size: Long, val extra: Map<String, Any> = emptyMap())(source)

Constructors

constructor(name: String, shape: Shape, dtype: String, offset: Long, size: Long, extra: Map<String, Any> = emptyMap())

Properties

val dtype: String

val extra: Map<String, Any>

val name: String

val offset: Long

val shape: Shape