cuequivariance-torch

张量积

EquivariantTensorProduct(e, *[, layout, ...])	等变张量积。
TensorProduct(descriptor, *[, device, ...])	PyTorch 模块，用于计算由描述符定义的分段张量积的最后一个操作数。

张量积的特殊情况

ChannelWiseTensorProduct(irreps_in1, irreps_in2)	通道式张量积层。
FullyConnectedTensorProduct(irreps_in1, ...)	全连接张量积层。
Linear(irreps_in, irreps_out, *[, layout, ...])	一个表示等变线性层的类。
SymmetricContraction(irreps_in, irreps_out, ...)	加速实现的对称收缩操作，在 https://arxiv.org/abs/2206.07697 中介绍。
TransposeIrrepsLayout(irreps, *, source, target)	转置张量的 irreps 布局。
SphericalHarmonics(ls[, normalize, device, ...])	将输入向量的球谐函数计算为 torch 模块。

欧几里得操作

Rotation(irreps, *[, layout, layout_in, ...])	一个表示 SO3 或 O3 表示的旋转层的类。
Inversion(irreps, *[, layout, layout_in, ...])	O(3) 表示的反演层。

encode_rotation_angle(angle, ell)	将角度编码为余弦和正弦张量。
vector_to_euler_angles(vector)	将 3D 向量转换为欧拉角。

额外模块

layers.BatchNorm(irreps, *[, layout, eps, ...])	正交表示的批量归一化。
layers.FullyConnectedTensorProductConv(...)	用于 DiffDock 类似架构中张量积的消息传递层。