Skip to content

GitLab

Unverified Commit c4f77c24 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix corner case in interpolate (#2146)

parent 3e60fb29
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/ops/misc.py
View file @ c4f77c24
...@@ -113,7 +113,7 @@ def interpolate(input, size=None, scale_factor=None, mode="nearest", align_corne ...@@ -113,7 +113,7 @@ def interpolate(input, size=None, scale_factor=None, mode="nearest", align_corne
) )
output_shape = _output_size(2, input, size, scale_factor) output_shape = _output_size(2, input, size, scale_factor)
output_shape = list(input.shape[:-2]) + output_shape output_shape = list(input.shape[:-2]) + list(output_shape)
return _new_empty_tensor(input, output_shape) return _new_empty_tensor(input, output_shape)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment