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) __rmul__() (GeometricTensor method) __sub__() (GeometricTensor method)

A

add_module() (SequentialModule method)

angle_offset (DiscretizationArgs attribute) append() (ModuleList method)

B

Basis (class in e2cnn.kernels) BasisExpansion (class in e2cnn.nn.modules.r2_conv) basisexpansion (R2Conv property) (R2Diffop property)

block_basisexpansion() (in module e2cnn.nn.modules.r2_conv) BlocksBasisExpansion (class in e2cnn.nn.modules.r2_conv) build_diffop_basis() (GeneralOnR2 method) build_kernel_basis() (GeneralOnR2 method) (GSpace method)

C

change_basis() (in module e2cnn.group) change_of_basis (FieldType property) change_of_basis_inv (FieldType property) character() (Representation method) check_equivariance() (EquivariantModule method) chi() (in module e2cnn.group.utils) combine() (CyclicGroup method) (DihedralGroup method) (Group method) (O2 method) (SO2 method)

ConcatenatedNonLinearity (class in e2cnn.nn) contains_trivial() (Representation method) cycle_isclose() (in module e2cnn.group.utils) cyclic_group() (in module e2cnn.group) CyclicGroup (class in e2cnn.group)

D

deltaorthonormal_init() (in module e2cnn.nn.init) DiffopBasis (class in e2cnn.diffops) diffops_CN_act_R2() (in module e2cnn.diffops) diffops_DN_act_R2() (in module e2cnn.diffops) diffops_Flip_act_R2() (in module e2cnn.diffops) diffops_O2_act_R2() (in module e2cnn.diffops) diffops_SO2_act_R2() (in module e2cnn.diffops)

diffops_Trivial_act_R2() (in module e2cnn.diffops) dihedral_group() (in module e2cnn.group) DihedralGroup (class in e2cnn.group) dimension() (BasisExpansion method) directsum() (in module e2cnn.group) DiscretizationArgs (class in e2cnn.diffops) disentangle() (in module e2cnn.group) DisentangleModule (class in e2cnn.nn)

E

e2cnn.diffops module e2cnn.group module e2cnn.group.utils module e2cnn.gspaces module e2cnn.kernels module e2cnn.nn module e2cnn.nn.init module ELU (class in e2cnn.nn) EmptyBasisException (class in e2cnn.kernels) equal() (CyclicGroup method) (DihedralGroup method) (Group method) (O2 method) (SO2 method) EquivariantModule (class in e2cnn.nn)

evaluate_output_shape() (EquivariantModule method) expand_parameters() (R2Conv method) (R2Diffop method) export() (ELU method) (EquivariantModule method) (GroupPooling method) (IdentityModule method) (InnerBatchNorm method) (ModuleList method) (PointwiseAdaptiveAvgPool method) (PointwiseAdaptiveMaxPool method) (PointwiseDropout method) (PointwiseMaxPool method) (PointwiseMaxPoolAntialiased method) (R2Conv method) (R2ConvTransposed method) (R2Diffop method) (R2Upsampling method) (ReLU method) (RestrictionModule method) (SequentialModule method) extend() (ModuleList method)

F

featurefield_action() (GSpace method) fibergroup (FieldType property) FieldDropout (class in e2cnn.nn) fields_end (FieldType property) fields_names() (FieldType method) fields_start (FieldType property) FieldType (class in e2cnn.nn) Flip2dOnR2 (class in e2cnn.gspaces) FlipRot2dOnR2 (class in e2cnn.gspaces) forward() (BasisExpansion method) (BlocksBasisExpansion method) (ELU method) (FieldDropout method) (GatedNonLinearity1 method) (GatedNonLinearity2 method) (GNormBatchNorm method) (GroupPooling method) (IdentityModule method) (InducedGatedNonLinearity1 method) (InducedNormBatchNorm method) (InducedNormNonLinearity method) (InducedNormPool method) (InnerBatchNorm method) (MultipleModule method) (NormBatchNorm method) (NormMaxPool method) (NormNonLinearity method) (NormPool method) (PointwiseAdaptiveAvgPool method) (PointwiseAdaptiveMaxPool method) (PointwiseAvgPool method) (PointwiseAvgPoolAntialiased method) (PointwiseDropout method) (PointwiseMaxPool method) (PointwiseMaxPoolAntialiased method) (PointwiseNonLinearity method) (R2Conv method) (R2Diffop method) (R2Upsampling method) (ReLU method) (SequentialModule method) (VectorFieldNonLinearity method)

G

GatedNonLinearity1 (class in e2cnn.nn) GatedNonLinearity2 (class in e2cnn.nn) GaussianRadialProfile (class in e2cnn.kernels) generalized_he_init() (in module e2cnn.nn.init) GeneralOnR2 (class in e2cnn.gspaces) GeometricTensor (class in e2cnn.nn) get_basis_info() (BasisExpansion method) get_basis_names() (BasisExpansion method)

get_dense_change_of_basis() (FieldType method) get_dense_change_of_basis_inv() (FieldType method) get_element_info() (BasisExpansion method) GNormBatchNorm (class in e2cnn.nn) Group (class in e2cnn.group) group_by_labels() (FieldType method) GroupPooling (class in e2cnn.nn) GSpace (class in e2cnn.gspaces)

I

IdentityModule (class in e2cnn.nn) index_select() (FieldType method) induced_repr() (GSpace method) induced_representation() (Group method) InducedGatedNonLinearity1 (class in e2cnn.nn) InducedNormBatchNorm (class in e2cnn.nn) InducedNormNonLinearity (class in e2cnn.nn) InducedNormPool (class in e2cnn.nn) InnerBatchNorm (class in e2cnn.nn) inverse() (CyclicGroup method) (DihedralGroup method) (Group method) (O2 method) (SO2 method)

IrreducibleRepresentation (class in e2cnn.group) irrep() (CyclicGroup method) (DihedralGroup method) (Group method) (GSpace method) (O2 method) (SO2 method) irreps (FieldType property) (Group property) (GSpace property) is_element() (Group method) is_trivial() (Representation method)

K

KernelBasis (class in e2cnn.kernels) kernels_CN_act_R2() (in module e2cnn.kernels) kernels_DN_act_R2() (in module e2cnn.kernels)

kernels_Flip_act_R2() (in module e2cnn.kernels) kernels_O2_act_R2() (in module e2cnn.kernels) kernels_SO2_act_R2() (in module e2cnn.kernels) kernels_Trivial_act_R2() (in module e2cnn.kernels)

L

LaplaceProfile (class in e2cnn.diffops)

load_cache() (in module e2cnn.diffops)

M

MaskModule (class in e2cnn.nn) MaxPoolChannels (class in e2cnn.nn) method (DiscretizationArgs attribute) module e2cnn.diffops e2cnn.group e2cnn.group.utils e2cnn.gspaces e2cnn.kernels e2cnn.nn e2cnn.nn.init

ModuleList (class in e2cnn.nn) MultipleModule (class in e2cnn.nn)

N

NormBatchNorm (class in e2cnn.nn) NormMaxPool (class in e2cnn.nn)

NormNonLinearity (class in e2cnn.nn) NormPool (class in e2cnn.nn)

O

O2 (class in e2cnn.group)

o2_group() (in module e2cnn.group) order() (Group method)

P

phi (DiscretizationArgs attribute) PointwiseAdaptiveAvgPool (class in e2cnn.nn) PointwiseAdaptiveMaxPool (class in e2cnn.nn) PointwiseAvgPool (class in e2cnn.nn) PointwiseAvgPoolAntialiased (class in e2cnn.nn) PointwiseDropout (class in e2cnn.nn)

PointwiseMaxPool (class in e2cnn.nn) PointwiseMaxPoolAntialiased (class in e2cnn.nn) PointwiseNonLinearity (class in e2cnn.nn) PolarBasis (class in e2cnn.kernels) pretty_print() (DiffopBasis method) psi() (in module e2cnn.group.utils) psichi() (in module e2cnn.group.utils)

Q

quotient_repr() (GSpace method)

quotient_representation() (Group method)

R

R2Conv (class in e2cnn.nn) R2ConvTransposed (class in e2cnn.nn) R2Diffop (class in e2cnn.nn) R2Upsampling (class in e2cnn.nn) regular_repr (GSpace property) regular_representation (Group property) ReLU (class in e2cnn.nn) Representation (class in e2cnn.group) representation (FieldType property) representations (Group property) (GSpace property)

ReshuffleModule (class in e2cnn.nn) restrict() (FieldType method) (Flip2dOnR2 method) (FlipRot2dOnR2 method) (GeometricTensor method) (GSpace method) (Representation method) (Rot2dOnR2 method) (TrivialOnR2 method) restrict_representation() (Group method) RestrictionModule (class in e2cnn.nn) Rot2dOnR2 (class in e2cnn.gspaces)

S

sample() (DiffopBasis method) (GaussianRadialProfile method) (KernelBasis method) (PolarBasis method) (SteerableKernelBasis method) sample_masked() (DiffopBasis method) (KernelBasis method) SequentialModule (class in e2cnn.nn) shape (GeometricTensor property) SingleBlockBasisExpansion (class in e2cnn.nn.modules.r2_conv) size() (GeometricTensor method) smoothing (DiscretizationArgs attribute)

SO2 (class in e2cnn.group) so2_group() (in module e2cnn.group) sorted() (FieldType method) split() (GeometricTensor method) SteerableDiffopBasis (class in e2cnn.diffops) SteerableKernelBasis (class in e2cnn.kernels) store_cache() (in module e2cnn.diffops) subgroup() (CyclicGroup method) (DihedralGroup method) (Group method) (O2 method) (SO2 method)

T

tensor_directsum() (in module e2cnn.nn) TensorBasis (class in e2cnn.diffops) testing_elements (FieldType property) (GSpace property) testing_elements() (CyclicGroup method) (DihedralGroup method) (Group method) (O2 method) (SO2 method)

to() (GeometricTensor method) train() (R2Conv method) (R2Diffop method) transform() (FieldType method) (GeometricTensor method) transform_fibers() (GeometricTensor method) trivial_group() (in module e2cnn.group) trivial_repr (GSpace property) trivial_representation (Group property) TrivialOnR2 (class in e2cnn.gspaces)

V

VectorFieldNonLinearity (class in e2cnn.nn)