bugfix: RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across 2528664 verified ZHANG886 commited on Jun 6