Function mean_pool

src/shape_assembly/models/encoder/vn_layers.py:184–185

(x, dim=-1, keepdim=False)

def mean_pool(x, dim=-1, keepdim=False):
    return x.mean(dim=dim, keepdim=keepdim)


class VNStdFeature(nn.Module):