Implementation of VNMaxPooling

[tranluan]

Hi @FlyingGiraffe,
In the maxpooling implementation, at this line:

vnn/models/vn_layers.py

Line 141 in 8e6a45b

idx = dotprod.max(dim=-1, keepdim=False)[1]

In order to get the indices idx, are we suppose to use argmax instead of max operation?

[FlyingGiraffe]

The output of the max function is a tuple of (values, indices) - in some later pytorch versions. If you're using an earlier version it may not return the indices