Type

struct Type

Full type of a compute-graph Value.

It combines an element DataType, a symbolic BatchSize and a static trailing shape.

The logical tensor shape is (batch_size, shape...). A batch_sizes value instead holds batch_size_list and has no element shape.

Public Functions

inline Type(DataType dtype, BatchSize batch_size, const std::vector<int> &shape)

Parameters:

• dtype – element data type

• batch_size – symbolic size of the batch dimension

• shape – static trailing dimensions

inline Type(const std::vector<BatchSize> &batch_size_list)

Construct a batch_sizes type carrying batch_size_list.

Public Members

DataType dtype

Element data type.

BatchSize batch_size

Symbolic size of the leading batch dimension.

std::vector<int> shape

Static trailing dimensions, without the batch dimension.

std::vector<BatchSize> batch_size_list

For a batch_sizes value: the list of sizes it carries.