e3x.nn.features

Utility functions for transforming features.

Functions

add(*inputs)	Add multiple input features.
change_max_degree_or_type(x[, max_degree, ...])	Changes the maximum degree and/or type of features.
even_degree_mask(max_degree)	Returns a mask than only keeps features with even degrees.
odd_degree_mask(max_degree)	Returns a mask than only keeps features with odd degrees.
reflect(x)	Reflects the input features into their mirror image.
rotate(x, wigner_d_matrix)	Rotates the input features with the given Wigner D-matrix (or -matrices).