def get_output_shape_for(self, input_shape):
output_shape = list(input_shape) # copy / convert to mutable list
output_shape[2] = pool_output_length(
input_shape[2],
pool_size=self.pool_size[0],
stride=self.stride[0],
pad=self.pad[0],
ignore_border=self.ignore_border,
)
output_shape[3] = pool_output_length(
input_shape[3],
pool_size=self.pool_size[1],
stride=self.stride[1],
pad=self.pad[1],
ignore_border=self.ignore_border,
)
output_shape[4] = pool_output_length(
input_shape[4],
pool_size=self.pool_size[2],
stride=self.stride[2],
pad=self.pad[2],
ignore_border=self.ignore_border,
)
return tuple(output_shape)
def get_output_shape_for(self, input_shape):
output_shape = list(input_shape) # copy / convert to mutable list
output_shape[2] = pool_output_length(input_shape[2],
pool_size=self.pool_size[0],
stride=self.stride[0],
pad=self.pad[0],
ignore_border=self.ignore_border,
)
output_shape[3] = pool_output_length(input_shape[3],
pool_size=self.pool_size[1],
stride=self.stride[1],
pad=self.pad[1],
ignore_border=self.ignore_border,
)
output_shape[4] = pool_output_length(input_shape[4],
pool_size=self.pool_size[2],
stride=self.stride[2],
pad=self.pad[2],
ignore_border=self.ignore_border,
)
return tuple(output_shape)
def get_output_shape_for(self, input_shape):
output_shape = list(input_shape) # copy / convert to mutable list
tr = len(output_shape) - self.n
for i in xrange(self.n):
output_shape[tr+i] = pool.pool_output_length(input_shape[tr+i],
pool_size=self.pool_size[i],
stride=self.stride[i],
pad=self.pad[i],
ignore_border=self.ignore_border,)
return tuple(output_shape)
def get_output_shape_for(self, input_shape):
output_shape = list(input_shape) # copy / convert to mutable list
tr = len(output_shape) - self.n
for i in xrange(self.n):
output_shape[tr + i] = pool.pool_output_length(
input_shape[tr + i],
pool_size=self.pool_size[i],
stride=self.stride[i],
pad=self.pad[i],
ignore_border=self.ignore_border,
)
return tuple(output_shape)