Index

_

__add__() (FieldType method) (GeometricTensor method) (Representation method) __call__() (Representation method) __getitem__() (GeometricTensor method) __iadd__() (GeometricTensor method) __imul__() (GeometricTensor method) __isub__() (GeometricTensor method)

__iter__() (FieldType method) __len__() (FieldType method) __mul__() (GeometricTensor method) __rmatmul__() (GeometricTensor method) __rmul__() (GeometricTensor method) __sub__() (GeometricTensor method) _RdConv (class in escnn.nn.modules.conv) _RdPointConv (class in escnn.nn.modules.pointconv)

A

add_module() (SequentialModule method)

AdjointBasis (class in escnn.kernels) append() (SequentialModule method)

B

basespace_action (GSpace property) (GSpace0D property) (GSpace2D property) (GSpace3D property) basis() (HomSpace method) basisexpansion (_RdConv property) (Linear property) (TensorProductModule property) BasisManager (class in escnn.nn.modules.basismanager) basissampler (_RdPointConv property) bl_induced_representation() (SO3 method) bl_irreps() (CyclicGroup method) (DihedralGroup method) (Icosahedral method) (O2 method) (O3 method) (SO2 method) (SO3 method)

bl_quotient_representation() (O2 method) (O3 method) (SO3 method) bl_regular_representation() (Icosahedral method) (O2 method) (O3 method) (SO2 method) (SO3 method) bl_sphere_representation() (O3 method) (SO3 method) BlocksBasisExpansion (class in escnn.nn.modules.basismanager) BlocksBasisSampler (class in escnn.nn.modules.basismanager) build_fiber_intertwiner_basis() (GSpace method) build_kernel_basis() (GSpace method) (GSpace0D method)

C

change_basis() (in module escnn.group) change_of_basis (FieldType property) change_of_basis_inv (FieldType property) character() (Representation method) check_equivariance() (EquivariantModule method) chi() (in module escnn.group.utils) CircularShellsBasis (class in escnn.kernels) complete_basis() (HomSpace method) compute_harmonics() (SteerableKernelBasis method)

compute_messages() (BlocksBasisSampler method) ConcatenatedNonLinearity (class in escnn.nn) conicalOnR3() (in module escnn.gspaces) contains_trivial() (Representation method) cycle_isclose() (in module escnn.group.utils) cyclic_group() (in module escnn.group) CyclicGroup (class in escnn.group) cylinder_discrete_group() (in module escnn.group) cylinder_group() (in module escnn.group) cylindricalOnR3() (in module escnn.gspaces)

D

deltaorthonormal_init() (in module escnn.nn.init) dihedral_group() (in module escnn.group) DihedralGroup (class in escnn.group) dihedralOnR3() (in module escnn.gspaces) dim_harmonic() (IrrepBasis method) (SteerableFiltersBasis method) (SteerableKernelBasis method) dimension() (BasisManager method)

dimension_basis() (HomSpace method) dimensionality (SteerableFiltersBasis property) direct_product() (in module escnn.group) DirectProductGroup (class in escnn.group) directsum() (in module escnn.group) disentangle() (in module escnn.group) DisentangleModule (class in escnn.nn) double_group() (in module escnn.group) DoubleGroup (class in escnn.group)

E

element() (Group method) elements (Group property) ELU (class in escnn.nn) EmptyBasisException (class in escnn.kernels) endomorphism_basis() (Representation method) EquivariantModule (class in escnn.nn) escnn.group module escnn.group.utils module escnn.gspaces module escnn.kernels module escnn.nn module escnn.nn.init module evaluate_output_shape() (EquivariantModule method) expand_bias() (_RdPointConv method) expand_filter() (_RdPointConv method) expand_parameters() (_RdConv method) (_RdPointConv method) (Linear method) (TensorProductModule method)

export() (ELU method) (EquivariantModule method) (GNormBatchNorm method) (GroupPooling method) (IdentityModule method) (IIDBatchNorm1d method) (IIDBatchNorm2d method) (IIDBatchNorm3d method) (InnerBatchNorm method) (LeakyReLU method) (Linear method) (PointwiseDropout method) (R2Conv method) (R2ConvTransposed method) (R3Conv method) (R3ConvTransposed method) (ReLU method) (RestrictionModule method) (SequentialModule method)

F

featurefield_action() (GSpace method) fiber_representation() (FieldType method) fibergroup (FieldType property) FieldDropout (class in escnn.nn) FieldNorm (class in escnn.nn) fields_end (FieldType property) fields_names() (FieldType method) fields_start (FieldType property) FieldType (class in escnn.nn) flip2dOnR2() (in module escnn.gspaces) flipRot2dOnR2() (in module escnn.gspaces) flipRot3dOnR3() (in module escnn.gspaces) flips_order (GSpace2D property) forward() (_RdConv method) (_RdPointConv method) (BlocksBasisExpansion method) (BlocksBasisSampler method) (ELU method) (FieldDropout method) (FieldNorm method) (FourierPointwise method) (GatedNonLinearity1 method) (GatedNonLinearity2 method) (GatedNonLinearityUniform method) (GNormBatchNorm method) (GroupPooling method) (IdentityModule method) (InducedGatedNonLinearity1 method) (InducedNormBatchNorm method) (InducedNormNonLinearity method) (InducedNormPool method) (InnerBatchNorm method) (KernelBasis method) (LeakyReLU method) (Linear method) (MultipleModule method) (NormBatchNorm method) (NormMaxPool method) (NormNonLinearity method) (NormPool method) (PointwiseAvgPool2D method) (PointwiseAvgPool3D method) (PointwiseAvgPoolAntialiased2D method) (PointwiseAvgPoolAntialiased3D method) (PointwiseDropout method) (PointwiseNonLinearity method) (QuotientFourierPointwise method) (R2Conv method) (R3Conv method) (ReLU method) (SequentialModule method) (VectorFieldNonLinearity method)

FourierELU (class in escnn.nn) FourierPointwise (class in escnn.nn) full_cylinder_discrete_group() (in module escnn.group) full_cylinder_group() (in module escnn.group) full_ico_group() (in module escnn.group) full_octa_group() (in module escnn.group) fullCylindricalOnR3() (in module escnn.gspaces) fullIcoOnR3() (in module escnn.gspaces) fullOctaOnR3() (in module escnn.gspaces)

G

GatedNonLinearity1 (class in escnn.nn) GatedNonLinearity2 (class in escnn.nn) GatedNonLinearityUniform (class in escnn.nn) GaussianRadialProfile (class in escnn.kernels) generalized_he_init() (in module escnn.nn.init) generators (Group property) GeometricTensor (class in escnn.nn) get_basis_info() (BasisManager method) get_dense_change_of_basis() (FieldType method) get_dense_change_of_basis_inv() (FieldType method) get_element_info() (BasisManager method) get_irrep_id() (Group method) GNormBatchNorm (class in escnn.nn) grid() (CyclicGroup method) (DihedralGroup method) (DirectProductGroup method) (Group method) (O2 method) (O3 method) (SO2 method) (SO3 method)

grid_so2() (O2 method) grid_so3() (O3 method) Group (class in escnn.group) group_by_labels() (FieldType method) GroupElement (class in escnn.group) GroupPooling (class in escnn.nn) GSpace (class in escnn.gspaces) GSpace0D (class in escnn.gspaces) GSpace2D (class in escnn.gspaces) GSpace3D (class in escnn.gspaces)

H

HarmonicPolynomialR3 (class in escnn.nn) has_same_coords() (GeometricTensor method)

homomorphism_space() (in module escnn.group) HomSpace (class in escnn.group) homspace() (Group method)

I

ico_group() (in module escnn.group) icoOnR3() (in module escnn.gspaces) Icosahedral (class in escnn.group) identity (Group property) IdentityModule (class in escnn.nn) IIDBatchNorm1d (class in escnn.nn) IIDBatchNorm2d (class in escnn.nn) IIDBatchNorm3d (class in escnn.nn) index_select() (FieldType method) induced_repr() (GSpace method) induced_representation() (Group method) (HomSpace method) InducedGatedNonLinearity1 (class in escnn.nn) InducedNormBatchNorm (class in escnn.nn) InducedNormNonLinearity (class in escnn.nn) InducedNormPool (class in escnn.nn) InnerBatchNorm (class in escnn.nn) inversion (O3 property)

invOnR3() (in module escnn.gspaces) IrreducibleRepresentation (class in escnn.group) irrep() (CyclicGroup method) (DihedralGroup method) (DirectProductGroup method) (Group method) (GSpace method) (Icosahedral method) (O2 method) (O3 method) (Octahedral method) (SO2 method) (SO3 method) IrrepBasis (class in escnn.kernels) irreps (FieldType property) (GSpace property) irreps() (Group method) is_compatible() (GeometricTensor method) is_trivial() (Representation method)

K

KernelBasis (class in escnn.kernels) kernels_CN_act_R2() (in module escnn.kernels) kernels_CN_act_R3() (in module escnn.kernels) kernels_DN_act_R2() (in module escnn.kernels) kernels_DN_conical_act_R3() (in module escnn.kernels) kernels_DN_dihedral_act_R3() (in module escnn.kernels) kernels_Flip_act_R2() (in module escnn.kernels) kernels_FullIco_act_R3() (in module escnn.kernels) kernels_FullOcta_act_R3() (in module escnn.kernels) kernels_FullTetra_act_R3() (in module escnn.kernels) kernels_Ico_act_R3() (in module escnn.kernels) kernels_Inv_act_R3() (in module escnn.kernels) kernels_O2_act_R2() (in module escnn.kernels) kernels_O2_conical_act_R3() (in module escnn.kernels) kernels_O2_dihedral_act_R3() (in module escnn.kernels)

kernels_O2_subgroup_act_R2() (in module escnn.kernels) kernels_O3_act_R3() (in module escnn.kernels) kernels_O3_subgroup_act_R3() (in module escnn.kernels) kernels_Octa_act_R3() (in module escnn.kernels) kernels_on_point() (in module escnn.kernels) kernels_Pyrito_act_R3() (in module escnn.kernels) kernels_SO2_act_R2() (in module escnn.kernels) kernels_SO2_act_R3() (in module escnn.kernels) kernels_SO2_subgroup_act_R2() (in module escnn.kernels) kernels_SO3_act_R3() (in module escnn.kernels) kernels_SO3_subgroup_act_R3() (in module escnn.kernels) kernels_Tetra_act_R3() (in module escnn.kernels) kernels_Trivial_act_R2() (in module escnn.kernels) kernels_Trivial_act_R3() (in module escnn.kernels) klein4_group() (in module escnn.group)

L

LeakyReLU (class in escnn.nn)

Linear (class in escnn.nn)

M

MaskModule (class in escnn.nn) MaxPoolChannels (class in escnn.nn) message() (_RdPointConv method) mirOnR3() (in module escnn.gspaces) module escnn.group escnn.group.utils escnn.gspaces escnn.kernels escnn.nn escnn.nn.init

MultipleModule (class in escnn.nn) multiplicity() (Representation method) (SteerableFiltersBasis method)

N

no_base_space() (in module escnn.gspaces) NormBatchNorm (class in escnn.nn)

NormMaxPool (class in escnn.nn) NormNonLinearity (class in escnn.nn) NormPool (class in escnn.nn)

O

O2 (class in escnn.group) o2_group() (in module escnn.group) O3 (class in escnn.group) o3_group() (in module escnn.group)

octa_group() (in module escnn.group) Octahedral (class in escnn.group) octaOnR3() (in module escnn.gspaces) order() (Group method)

P

PARAM (Group property) param (GroupElement property) PARAMETRIZATIONS (Group property) PointBasis (class in escnn.kernels) PointwiseAdaptiveAvgPool2D (class in escnn.nn) PointwiseAdaptiveAvgPool3D (class in escnn.nn) PointwiseAdaptiveMaxPool2D (class in escnn.nn) PointwiseAdaptiveMaxPool3D (class in escnn.nn) PointwiseAvgPool2D (class in escnn.nn) PointwiseAvgPool3D (class in escnn.nn)

PointwiseAvgPoolAntialiased2D (class in escnn.nn) PointwiseAvgPoolAntialiased3D (class in escnn.nn) PointwiseDropout (class in escnn.nn) PointwiseMaxPool2D (class in escnn.nn) PointwiseMaxPool3D (class in escnn.nn) PointwiseMaxPoolAntialiased2D (class in escnn.nn) PointwiseMaxPoolAntialiased3D (class in escnn.nn) PointwiseNonLinearity (class in escnn.nn) psi() (in module escnn.group.utils) psichi() (in module escnn.group.utils)

Q

quotient_repr() (GSpace method) quotient_representation() (Group method)

QuotientFourierELU (class in escnn.nn) QuotientFourierPointwise (class in escnn.nn)

R

R2Conv (class in escnn.nn) R2ConvTransposed (class in escnn.nn) R2PointConv (class in escnn.nn) R2Upsampling (class in escnn.nn) R3Conv (class in escnn.nn) R3ConvTransposed (class in escnn.nn) R3IcoConv (class in escnn.nn) R3IcoConvTransposed (class in escnn.nn) R3PointConv (class in escnn.nn) R3Upsampling (class in escnn.nn) regular_repr (GSpace property) regular_representation (Group property) ReLU (class in escnn.nn) Representation (class in escnn.group) representation (FieldType property) representations (Group property) (GSpace property)

ReshuffleModule (class in escnn.nn) restrict() (FieldType method) (GeometricTensor method) (GSpace method) (GSpace0D method) (GSpace2D method) (GSpace3D method) (Representation method) restrict_representation() (Group method) RestrictedWignerEckartBasis (class in escnn.kernels) RestrictionModule (class in escnn.nn) rot2dOnR2() (in module escnn.gspaces) rot2dOnR3() (in module escnn.gspaces) rot3dOnR3() (in module escnn.gspaces) rotations_order (GSpace2D property)

S

sample() (AdjointBasis method) (CircularShellsBasis method) (GaussianRadialProfile method) (Group method) (IrrepBasis method) (KernelBasis method) (SparseOrbitBasis method) (SphericalShellsBasis method) (SteerableFiltersBasis method) (SteerableKernelBasis method) sample_as_dict() (SteerableFiltersBasis method) sample_harmonics() (IrrepBasis method) (SteerableKernelBasis method) scalar_basis() (HomSpace method) SequentialModule (class in escnn.nn) shape (GeometricTensor property) size() (GeometricTensor method) SO2 (class in escnn.group) so2_group() (in module escnn.group) SO3 (class in escnn.group) so3_group() (in module escnn.group) sorted() (FieldType method) SparseOrbitBasis (class in escnn.kernels)

SparseOrbitBasisWithIcosahedralSymmetry (class in escnn.kernels) spectral_quotient_representation() (Group method) spectral_regular_representation() (Group method) sphere_grid() (O3 method) (SO3 method) SphericalShellsBasis (class in escnn.kernels) split() (GeometricTensor method) standard_representation (Icosahedral property) (Octahedral property) standard_representation() (O2 method) (O3 method) (SO2 method) (SO3 method) SteerableFiltersBasis (class in escnn.kernels) SteerableKernelBasis (class in escnn.kernels) subgroup() (Group method) subgroup1_id (DirectProductGroup property) subgroup2_id (DirectProductGroup property) subgroup_diagonal_id (DoubleGroup property) subgroup_self_id (DirectProductGroup property) (Group property) subgroup_trivial_id (DirectProductGroup property) (Group property)

T

tensor() (Representation method) tensor_directsum() (in module escnn.nn) TensorProductModule (class in escnn.nn) testing_elements (FieldType property) (GSpace property) testing_elements() (CyclicGroup method) (DihedralGroup method) (DirectProductGroup method) (Group method) (Icosahedral method) (O2 method) (O3 method) (Octahedral method) (SO2 method) (SO3 method) to() (GeometricTensor method) (GroupElement method)

train() (_RdConv method) (_RdPointConv method) (Linear method) (TensorProductModule method) transform() (FieldType method) (GeometricTensor method) transform_fibers() (FieldType method) (GeometricTensor method) trivial_group() (in module escnn.group) trivial_repr (GSpace property) trivial_representation (DirectProductGroup property) (Group property) trivialOnR2() (in module escnn.gspaces) trivialOnR3() (in module escnn.gspaces) type() (GSpace method)

U

uniform (FieldType property)

UnionBasis (class in escnn.kernels)

V

value (GroupElement property)

VectorFieldNonLinearity (class in escnn.nn)

W

WignerEckartBasis (class in escnn.kernels)