struct

`SushiAI::IO::TensorRecord`

One tensor's identity in a checkpoint: its name, type and extents.

Declared in: include/SushiAI/io/checkpoint_format.hpp

The shape is recorded rather than an element count, and that is the difference between a checkpoint that catches a transposed weight and one that loads [256, 6] happily into [6, 256].

Public attributes

std::string name

The dotted parameter name the walk produced, e.g.

"0.weight".

DType dtype = DType::FLOAT32

The element type.

Shape shape {}

The extents.

Public member functions

std::size_t payload_bytes() const noexcept

How many bytes this record's payload occupies.

Returns

The element count times the element size.