Update onnxscript/function_libs/torch_lib/ops/nn.py

Co-authored-by: Justin Chu <[email protected]>
microsoft · Aug 5, 2024 · 601f5d3 · 601f5d3
1 parent 41200fc
commit 601f5d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py
@@ -672,7 +672,7 @@ def _get_im2col_indices_along_dim(
     # each dimension d ranges from 0 to input[d]+2xpadding[d]-dilation[d]x(kernel_size[d]-1)
     # with steps = stride
 
-    blocks_d = input_d + (padding_d * 2) - (dilation_d * (kernel_size_d - 1))
+    blocks_d = input_d + ((padding_d * 2) - (dilation_d * (kernel_size_d - 1)))
 
     # Stride kernel over input and find starting indices along dim d
     blocks_d_indices = op.Range(0, blocks_d, stride_d)