TensorDescriptor

lamp.data.schemas.TensorDescriptor
See theTensorDescriptor companion object
case class TensorDescriptor(dims: Seq[Long], dataType: Byte, byteOffset: Long, byteLength: Long)

dataType is pytorch scalartype:

  • 1 I8
  • 2 I16
  • 3 I32
  • 4 I64
  • 5 FP16
  • 6 FP32
  • 7 FP64
  • 11 bool
  • 16 BF16

Attributes

Companion
object
Graph
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all

Members list

Value members

Inherited methods

def productElementNames: Iterator[String]

Attributes

Inherited from:
Product
def productIterator: Iterator[Any]

Attributes

Inherited from:
Product