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)
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)
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)