[Bug]: keepdim not working for any/all when axis=None

[neosunhan]

What happened?

axis=None in any/all performs a reduction across all dimensions of the array. However, keepdim=True does not preserve shape in this case.

Code snippet triggering the error

>>> import heat as ht
>>> a = ht.ones((1,1))
>>> a.shape
(1, 1)
>>> b = ht.all(a, keepdim=True)
>>> b.shape
(1,)

Version

main (development branch)

Python version

3.10

PyTorch version

1.11