about the patch_partition

[FairyYukikaze]

When training a classification task, I encountered the following error: RuntimeError: Trying to create tensor with negative dimension -2147486105: [-2147486105]. Why would num be a negative value?

[Aristo23333]

When training a classification task, I encountered the following error: RuntimeError: Trying to create tensor with negative dimension -2147486105: [-2147486105]. Why would num be a negative value?

Hi! This issue is due to a division by zero error. You can recover it by add this line. Sorry for my original code containing this bug.