nir.ir.flatten

Contents

`nir.ir.flatten`#

Module Contents#

Classes#

Flatten node.

class nir.ir.flatten.Flatten#

Bases: nir.ir.node.NIRNode

Flatten node.

This node flattens its input tensor. input_type must be a dict with one key: “input”.

input_type: nir.ir.typing.Types#

start_dim: int = 1#

end_dim: int#

input_type: Optional[Dict[str, numpy.ndarray]]#

output_type: Optional[Dict[str, numpy.ndarray]]#

metadata: Dict[str, Any]#

__post_init__()#

to_dict() → Dict[str, Any]#

Serialize into a dictionary.

Return type: Dict[str, Any]

classmethod from_dict(node: Dict[str, Any])#

Parameters: node (Dict[str, Any]) –