Skip to content

Commit

Permalink
Fix lint failures
Browse files Browse the repository at this point in the history
  • Loading branch information
shubhambhokare1 committed Apr 5, 2021
1 parent 6acd9d7 commit 3e193da
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torch/onnx/symbolic_opset11.py
Expand Up @@ -881,12 +881,12 @@ def repeat_interleave(g, self, repeats, dim=None):
reps = sym_help._size_helper(g, input, dim)
reps = unsqueeze(g, reps, 0)
repeats = g.op("Expand", repeats, reps)
# There are cases when the repeats are 1-d tensor with multiple repeats, but dim
# There are cases when the repeats are 1-d tensor with multiple repeats, but dim
# provided along one of the dynamic axes provided. A simple example would be
# input.shape -> [1, 1, *] where * represents the dynamic axes, and dim = 2
# Now, repeat interleaving can be performed in pytorch when the value of * matches
# with the number of elements in repeat, for example if * -> 2, number of repeats
# should be 2 as well.
# should be 2 as well.
else:
return torch.onnx.symbolic_opset9.repeat_interleave(g, self, repeats, final_dim)

Expand Down

0 comments on commit 3e193da

Please sign in to comment.