jittor.nn.AdaptiveMaxPool3d会对合规的输入报错

[PhyllisJi]

Describe the bug

当输入一个正常的张量给AdaptiveMaxPool3d层，其中的第499行会对该输入报错并停止运算

Full Log

ZeroDivisionError
File ~\AppData\Local\Programs\Python\Python310\Lib\site-packages\jittor\
_init__py:1168,in Module._call__(self,*args, **kw)
def __call__(self, *args, **kw):->
return self.execute(*args, **kw)
File~\AppData\Local\Programs\Python\Python310\Lib\site-paokages\jittor\pool.py:499,in AdaptiveMaxPool3d.execute(self. x)
if self.return_indices:
MaxPool3dC
return M 
kerneL_size=(self.ksd,self.ksh,self.ksw),
stride=(self.sd,self.sh,self.sw),return_indices=True)(x)
d= (D-self.ksd)//self.sd+1 500 h = (H-self.ksh)//self.sh+1 501 w = (W-self.ksw)//self.sw+1
ZeroDivisionError:integer division or modulo by zero.

Minimal Reproduce

p = jittor.nn.AdaptiveMaxPool3d(output_size = 6)
x = jittor.randn(209952,4,4,4,4)
y = p(x)

Expected behavior

正常运算，获取结果