Skip to content

split

torch_to_nnef.op.aten.split

chunk

chunk(g, node, name_to_tensor, **kwargs)

Map PyTorch: 'aten:chunk' to NNEF.

split_with_sizes

split_with_sizes(g, node, name_to_tensor, **kwargs)

Translate aten::split_with_sizes to NNEF.

We are aware that. split<?>( value: tensor<?>, axis: integer, ratios: integer[] ) -> ( values: tensor<?>[] )

exists but since tract does not support it, we reexpress it with slice

unbind

unbind(g, node, name_to_tensor, **kwargs)

Unbind is unstack in NNEF.