Shape mismatch error

It seems the error message might be raised from F.interpolate and you should thus check the provided size as well as the shape of the input tensor.
This topic discusses the same issue.

1 Like