Skip to content

concat

torch_to_nnef.op.aten.concat

cat

cat(g, node, name_to_tensor, torch_graph, **kwargs)

Map PyTorch: 'aten:cat' to NNEF.

hstack

hstack(g, node, name_to_tensor, torch_graph, **kwargs)

Map PyTorch: 'aten:hstack' to NNEF.

roll

roll(g, node, name_to_tensor, inference_target, **kwargs)

Map PyTorch: 'aten:roll' to NNEF.

stack

stack(g, node, name_to_tensor, torch_graph, **kwargs)

Map PyTorch: 'aten:stack' to NNEF.

vstack

vstack(g, node, name_to_tensor, torch_graph, **kwargs)

Map PyTorch: 'aten:vstack' to NNEF.