Skip to content

StateMetadata

Full path: schola.core.model.StateMetadata

Metadata for recurrent or sequential state tensors in ONNX export.

StateMetadata(has_seq_dim = False, max_seq_len = None, seq_dim = None)

Parameters

  • has_seq_dim (bool)

  • max_seq_len (int)

  • seq_dim (int)

Methods

init

__init__(has_seq_dim = False, max_seq_len = None, seq_dim = None)

Parameters

  • has_seq_dim (bool)

  • max_seq_len (int)

  • seq_dim (int)


to_dict

to_dict()

Serialize state metadata to string values for ONNX metadata_props.

Returns

Keys has_seq_dim and, when sequential, max_seq_len and seq_dim.

Attributes

has_seq_dim

has_seq_dim

max_seq_len

max_seq_len

seq_dim

seq_dim