Python unbind Examples

Example #1

def _unbind_helper(g, self, dim, _outputs):
    if _export_onnx_opset_version < 11:
        from torch.onnx.symbolic_opset9 import unbind
    elif _export_onnx_opset_version <= 12:
        from torch.onnx.symbolic_opset11 import unbind  # type: ignore[no-redef]
    else:
        from torch.onnx.symbolic_opset13 import unbind  # type: ignore[no-redef]
    return unbind(g, self, dim, _outputs)

Example #2

def unbind(g, self, dim=0, _outputs=None):
    if _outputs is None:
        return g.op(
            "SplitToSequence",
            self,
            g.op("Constant", value_t=torch.tensor(1, dtype=torch.long)),
            axis_i=dim,
            keepdims_i=0,
        )
    else:
        return opset9.unbind(g, self, dim, _outputs)

Example #3

File: symbolic_helper.py Project: mingzhe09088/pytorch-1

def _unbind_helper(g, self, dim, _outputs):
    if _export_onnx_opset_version <= 9:
        from torch.onnx.symbolic_opset9 import unbind
    else:
        from torch.onnx.symbolic_opset11 import unbind
    return unbind(g, self, dim, _outputs)