@@ -36,7 +36,7 @@ def symb_forward(self, symb_input):
3636
3737 # downsample height and width first
3838 # other dimensions contribute to batch_size
39- op = Pool (self .window_size [1 :], self .ignore_border , st = self .stride [1 :], padding = self .padding [1 :])
39+ op = Pool (self .window_size [1 :], self .ignore_border , stride = self .stride [1 :], pad = self .padding [1 :])
4040 output = op (input_4d )
4141
4242 outshape = df .T .join (0 , symb_input .shape [:- 2 ], output .shape [- 2 :])
@@ -56,7 +56,7 @@ def symb_forward(self, symb_input):
5656
5757 # downsample depth
5858 # other dimensions contribute to batch_size
59- op = Pool ((1 ,self .window_size [0 ]), self .ignore_border , st = (1 ,self .stride [0 ]), padding = (0 , self .padding [0 ]))
59+ op = Pool ((1 ,self .window_size [0 ]), self .ignore_border , stride = (1 ,self .stride [0 ]), pad = (0 , self .padding [0 ]))
6060 outdepth = op (input_4D_depth )
6161
6262 outshape = df .T .join (0 , input_depth .shape [:- 2 ], outdepth .shape [- 2 :])
@@ -66,8 +66,8 @@ def symb_forward(self, symb_input):
6666 else :
6767 return pool_2d (
6868 symb_input ,
69- ds = self .window_size ,
69+ ws = self .window_size ,
7070 ignore_border = self .ignore_border ,
71- st = self .stride ,
72- padding = self .padding
71+ stride = self .stride ,
72+ pad = self .padding
7373 )
0 commit comments