Class ModelMetadataTensor

Class Documentation

class ModelMetadataTensor

This class holds the metadata associated with an input tensor.

Public Functions

ModelMetadataTensor(const std::string &name, DataType datatype, std::vector<uint64_t> shape)

Construct a new Model Metadata Tensor object.

Parameters

  • name – name of the tensor

  • datatype – the datatype this tensor accepts

  • shape – the expected shape of the data

const std::string &getName() const

Gets the name of the tensor.

const DataType &getDataType() const

Gets the datatype that this tensor accepts.

const std::vector<uint64_t> &getShape() const

Gets the expected shape of the data.