Implement symbolic for slice operation (#5204)

sungin-h · Feb 13, 2018 · f96f3c3 · f96f3c3
1 parent ab18aae
commit f96f3c3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torch/onnx/symbolic.py b/torch/onnx/symbolic.py
@@ -534,6 +534,12 @@ def conv_tbc(g, input, weight, bias, pad):
     return g.op("ATen", input, weight, bias, operator_s="conv_tbc", pad_i=pad)
 
 
+def slice(g, self, dim, start, end, step):
+    if step != 1:
+        _unimplemented("slice", "step!=1 is currently not supported")
+    return g.op("Slice", self, axes_i=[dim], starts_i=[start], ends_i=[end])
+
+
 def instance_norm(g, input, **kwargs):
     input_type = input.type().scalarType()
     weight = kwargs.get("weight", None)