e3x.nn.features.reflect

e3x.nn.features.reflect(x)[source]

Reflects the input features into their mirror image.

Parameters:: x (Union[Float[Array, '... 1 (max_degree+1)**2 num_features'], Float[Array, '... 2 (max_degree+1)**2 num_features']]) – Input features with shape (…, S, (L+1)**2, F).
Return type:: Union[Float[Array, '... 1 (max_degree+1)**2 num_features'], Float[Array, '... 2 (max_degree+1)**2 num_features']]
Returns:: The reflected features (with the same shape as x).