宽度超过 512 会报错view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

[gogoclaire]

在宽度大于512 时会报错，提示：view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.
请问这个要做怎么解决？