Skip to main content

Python class

TensorSpec

TensorSpec​

class max.engine.TensorSpec

source

Bases: object

Defines the properties of a tensor, including its name, shape and data type.

For usage examples, see Model.input_metadata.

dtype​

property dtype

source

A tensor data type.

name​

property name

source

A tensor name.

shape​

property shape

source

The shape of the tensor as a list of integers.

If a dimension size is unknown/dynamic (such as the batch size), its value is None.