Bugs in pool when stride is odd

[fangwei123456]

Objective of issue:

Lava DL version:

• 0.3.0 (feature release)
• 0.2.1 (bug fixes)
• 0.2.0 (current version)
• 0.1.2

Lava version:

• 0.4.0 (feature release)
• 0.3.1 (bug fixes)
• 0.3.0 (current version)
• 0.2.0
• 0.1.2

I'm submitting a ...

• bug report
• feature request
• documentation request

Current behavior:

• Raise an error

  File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/lava/lib/dl/slayer/synapse/layer.py", line 502, in forward
    return result.reshape((
RuntimeError: shape '[2, 4, -1, 8, 4]' is invalid for input of size 2112

Expected behavior:

• Output correct results

Steps to reproduce:

x = torch.rand([2, 4, 16, 16, 4])
pool = slayer.synapse.Pool(kernel_size=3, stride=2)
pool(x)

Related code:

insert short code snippets here

Other information:

(lava-env) wfang@mlg-ThinkStation-P920:~$ conda list la
# packages in environment at /home/wfang/anaconda3/envs/lava-env:
#
# Name                    Version                   Build  Channel
lava                      0.4.0              pyhd8ed1ab_0    conda-forge
lava-dl                   0.2.0              pyhd8ed1ab_0    conda-forge

[pbonazzi]

The same error happens when the input shape (width and height) are odd.