Index

A

Activation (class in e3nn.nn) angles_to_axis_angle() (in module e3nn.o3) angles_to_matrix() (in module e3nn.o3) angles_to_quaternion() (in module e3nn.o3) angles_to_xyz() (in module e3nn.o3)

assert_auto_jitable() (in module e3nn.util.test) assert_equivariant() (in module e3nn.util.test) assert_normalized() (in module e3nn.util.test) axis_angle_to_angles() (in module e3nn.o3) axis_angle_to_matrix() (in module e3nn.o3) axis_angle_to_quaternion() (in module e3nn.o3)

B

BatchNorm (class in e3nn.nn)

C

CartesianTensor (class in e3nn.io) change_of_basis (e3nn.o3.ReducedTensorProducts attribute) compile() (in module e3nn.util.jit) compile_mode() (in module e3nn.util.jit) complete_basis() (in module e3nn.math) Compose (class in e3nn.nn.models.gate_points_2101)

compose_angles() (in module e3nn.o3) compose_axis_angle() (in module e3nn.o3) compose_quaternion() (in module e3nn.o3) Convolution (class in e3nn.nn.models.gate_points_2101) count() (e3nn.o3.Irrep method) (e3nn.o3.Irreps method)

D

D_from_angles() (e3nn.o3.Irrep method) (e3nn.o3.Irreps method) D_from_axis_angle() (e3nn.o3.Irrep method) (e3nn.o3.Irreps method) D_from_matrix() (e3nn.o3.Irrep method) (e3nn.o3.Irreps method)

D_from_quaternion() (e3nn.o3.Irrep method) (e3nn.o3.Irreps method) dim (e3nn.o3.Irrep property) (e3nn.o3.Irreps attribute) direct_sum() (in module e3nn.math)

E

e3nn.nn.models.gate_points_2101 module e3nn.util.jit module

e3nn.util.test module ElementwiseTensorProduct (class in e3nn.o3) equivariance_error() (in module e3nn.util.test)

F

find_peaks() (e3nn.io.SphericalTensor method) format_equivariance_error() (in module e3nn.util.test) forward() (e3nn.nn.Activation method) (e3nn.nn.BatchNorm method) (e3nn.nn.Gate method) (e3nn.nn.models.gate_points_2101.Compose method) (e3nn.nn.models.gate_points_2101.Convolution method) (e3nn.nn.models.gate_points_2101.Network method) (e3nn.nn.NormActivation method) (e3nn.nn.S2Activation method) (e3nn.o3.FromS2Grid method) (e3nn.o3.ReducedTensorProducts method) (e3nn.o3.TensorProduct method) (e3nn.o3.TensorSquare method) (e3nn.o3.ToS2Grid method)

from_cartesian() (e3nn.io.CartesianTensor method) from_samples_on_s2() (e3nn.io.SphericalTensor method) from_vectors() (e3nn.io.CartesianTensor method) FromS2Grid (class in e3nn.o3) FullTensorProduct (class in e3nn.o3) FullyConnectedNet (class in e3nn.nn) FullyConnectedTensorProduct (class in e3nn.o3)

G

Gate (class in e3nn.nn) get_compile_mode() (in module e3nn.util.jit)

get_tracing_inputs() (in module e3nn.util.jit) grid (e3nn.o3.FromS2Grid attribute) (e3nn.o3.ToS2Grid attribute)

I

identity_angles() (in module e3nn.o3) identity_quaternion() (in module e3nn.o3) index() (e3nn.o3.Irrep method) (e3nn.o3.Irreps method) inverse_angles() (in module e3nn.o3) inverse_quaternion() (in module e3nn.o3) irfft() (in module e3nn.o3)

Irrep (class in e3nn.o3) Irreps (class in e3nn.o3) irreps_in (e3nn.nn.Gate property) (e3nn.o3.ReducedTensorProducts attribute) irreps_out (e3nn.nn.Gate property) (e3nn.o3.ReducedTensorProducts attribute) is_scalar() (e3nn.o3.Irrep method) iterator() (e3nn.o3.Irrep class method)

L

l (e3nn.o3.Irrep property) Legendre() (in module e3nn.o3)

lmax (e3nn.o3.Irreps attribute) ls (e3nn.o3.Irreps attribute)

M

matrix_to_angles() (in module e3nn.o3) matrix_to_axis_angle() (in module e3nn.o3) matrix_to_quaternion() (in module e3nn.o3) matrix_x() (in module e3nn.o3) matrix_y() (in module e3nn.o3)

matrix_z() (in module e3nn.o3) module e3nn.nn.models.gate_points_2101 e3nn.util.jit e3nn.util.test

N

Network (class in e3nn.nn.models.gate_points_2101) NormActivation (class in e3nn.nn)

norms() (e3nn.io.SphericalTensor method) num_irreps (e3nn.o3.Irreps attribute)

O

orthonormalize() (in module e3nn.math)

P

p (e3nn.o3.Irrep property)

plot() (e3nn.io.SphericalTensor method) plotly_surface() (e3nn.io.SphericalTensor method)

Q

quaternion_to_angles() (in module e3nn.o3)

quaternion_to_axis_angle() (in module e3nn.o3) quaternion_to_matrix() (in module e3nn.o3)

R

rand_angles() (in module e3nn.o3) rand_axis_angle() (in module e3nn.o3) rand_matrix() (in module e3nn.o3) rand_quaternion() (in module e3nn.o3) randn() (e3nn.o3.Irreps method)

random_irreps() (in module e3nn.util.test) reduced_tensor_products() (e3nn.io.CartesianTensor method) ReducedTensorProducts (class in e3nn.o3) remove_zero_multiplicities() (e3nn.o3.Irreps method) rfft() (in module e3nn.o3) right() (e3nn.o3.TensorProduct method)

S

s2_grid() (in module e3nn.o3) S2Activation (class in e3nn.nn) script() (in module e3nn.util.jit) set_random_seeds() (in module e3nn.util.test) signal_on_grid() (e3nn.io.SphericalTensor method) signal_xyz() (e3nn.io.SphericalTensor method) simplify() (e3nn.o3.Irreps method) slices() (e3nn.o3.Irreps method)

soft_one_hot_linspace() (in module e3nn.math) soft_unit_step() (in module e3nn.math) sort() (e3nn.o3.Irreps method) spherical_harmonics() (e3nn.o3.Irreps static method) (in module e3nn.o3) spherical_harmonics_alpha_beta() (in module e3nn.o3) spherical_harmonics_s2_grid() (in module e3nn.o3) SphericalTensor (class in e3nn.io) sum_of_diracs() (e3nn.io.SphericalTensor method)

T

TensorProduct (class in e3nn.o3) TensorSquare (class in e3nn.o3) to_cartesian() (e3nn.io.CartesianTensor method)

ToS2Grid (class in e3nn.o3) trace() (in module e3nn.util.jit) trace_module() (in module e3nn.util.jit)

V

visualize() (e3nn.o3.TensorProduct method)

W

weight_view_for_instruction() (e3nn.o3.TensorProduct method) weight_views() (e3nn.o3.TensorProduct method)

wigner_3j() (in module e3nn.o3) wigner_D() (in module e3nn.o3) with_peaks_at() (e3nn.io.SphericalTensor method)

X

xyz_to_angles() (in module e3nn.o3)