mindscience.e3nn

nn

mindscience.e3nn.nn.Activation	标量不可约表示（l=0）的激活函数。
mindscience.e3nn.nn.BatchNorm	针对正交归一化群表示的批归一化。
mindscience.e3nn.nn.FullyConnectedNet	带有标量归一化激活的全连接神经网络。
mindscience.e3nn.nn.Gate	门控激活函数。
mindscience.e3nn.nn.NormActivation	不可约表示的范数的激活函数。
mindscience.e3nn.nn.OneHot	支持不可约表示注解的 one-hot 嵌入。
mindscience.e3nn.nn.Scatter	易用的 scatter 操作封装：根据索引将源张量的值聚合到目标张量中。
mindscience.e3nn.nn.soft_one_hot_linspace	投影到函数基上。
mindscience.e3nn.nn.soft_unit_step	单位阶跃函数的平滑版本。
mindscience.e3nn.nn.SoftOneHotLinspace	投影到函数基上。

o3

mindscience.e3nn.o3.angles_to_matrix	将欧拉角 \((\alpha, \beta, \gamma)\) 转换为对应的 \(3 \times 3\) 旋转矩阵。
mindscience.e3nn.o3.angles_to_xyz	将两个球面角 \((\alpha, \beta)\) 转换为单位球面上的笛卡尔坐标 \((x, y, z)\)。
mindscience.e3nn.o3.change_basis_real_to_complex	将实值球谐函数基转换为其复值对应基。
mindscience.e3nn.o3.compose_angles	计算由两个旋转组合而成的欧拉角。
mindscience.e3nn.o3.ElementwiseTensorProduct	元素级连接张量积。
mindscience.e3nn.o3.FullTensorProduct	两个不可约表示之间的完全张量积。
mindscience.e3nn.o3.FullyConnectedTensorProduct	全连接加权张量积。
mindscience.e3nn.o3.identity_angles	返回对应于“不旋转”的欧拉角 \((\alpha, \beta, \gamma)\) 的单位集合。
mindscience.e3nn.o3.Irrep	O(3) 的不可约表示。
mindscience.e3nn.o3.Irreps	O(3) 的不可约表示的直和。
mindscience.e3nn.o3.Linear	线性等变操作。
mindscience.e3nn.o3.LinearBias	具有添加偏置选项的线性等变操作。
mindscience.e3nn.o3.matrix_to_angles	将 \(3 \times 3\) 旋转矩阵转换为欧拉角 \((\alpha, \beta, \gamma)\)。
mindscience.e3nn.o3.matrix_x	返回绕 x 轴旋转给定角度的 \(3 \times 3\) 旋转矩阵。
mindscience.e3nn.o3.matrix_y	返回绕 y 轴旋转给定角度的 \(3 \times 3\) 旋转矩阵。
mindscience.e3nn.o3.matrix_z	返回绕 z 轴旋转给定角度的 \(3 \times 3\) 旋转矩阵。
mindscience.e3nn.o3.Norm	计算直和张量中每个不可约表示（irrep）的范数（长度）。
mindscience.e3nn.o3.rand_angles	返回一组均匀随机的欧拉角 \((\alpha, \beta, \gamma)\)，对应三维空间中的随机旋转。
mindscience.e3nn.o3.so3_generators	计算so(3)李代数生成元。
mindscience.e3nn.o3.spherical_harmonics	计算球面谐波。
mindscience.e3nn.o3.SphericalHarmonics	将三维笛卡尔向量 (x, y, z) 映射到复值球面谐波基函数 \(Y_l^m(\hat{x})\) 的层。
mindscience.e3nn.o3.su2_generators	计算su(2)李代数生成元。
mindscience.e3nn.o3.TensorProduct	多功能张量乘积运算符，适用于两个输入 Irreps 和一个输出 Irreps，将两个张量发送到一个张量中并保持几何张量属性。
mindscience.e3nn.o3.TensorSquare	计算张量的平方张量积。
mindscience.e3nn.o3.wigner_3j	Wigner 3j符号 \(C_{lmn}\)。
mindscience.e3nn.o3.wigner_D	SO(3) 的 Wigner D 矩阵表示。
mindscience.e3nn.o3.xyz_to_angles	将单位球面上的点 \(\vec r = (x, y, z)\) 转换为角度 \((\alpha, \beta)\)。

so2_conv

mindscience.e3nn.so2_conv.SO3Rotation	处理球面谐波不可约表示的 SO(3) 旋转的类。
mindscience.e3nn.so2_conv.SO2Convolution	面向圆群的复值特征的 SO(2) 等变卷积层。
mindscience.e3nn.so2_conv.init_edge_rot_mat	根据边距向量初始化旋转矩阵。

utils

mindscience.e3nn.utils.Ncon	多个张量的缩并运算符，功能类似于 Einsum。
mindscience.e3nn.utils.radius	为 y 中每个元素在距离 r 内找到 x 中的所有点。
mindscience.e3nn.utils.radius_full	为 y 中每个元素找到 x 中的所有点。
mindscience.e3nn.utils.radius_graph	计算给定距离内图所有点之间的边。
mindscience.e3nn.utils.radius_graph_full	计算给定距离内图所有点之间的边。