反演#
- class cuequivariance_torch.Inversion(
- irreps: Irreps,
- *,
- layout: IrrepsLayout | None = None,
- layout_in: IrrepsLayout | None = None,
- layout_out: IrrepsLayout | None = None,
- device: device | None = None,
- math_dtype: dtype | None = None,
- use_fallback: bool | None = None,
用于 \(O(3)\) 表示的反演层。
- 参数:
irreps (Irreps) – 要反演的张量的不可约表示。
layout (IrrepsLayout, optional) – 张量的内存布局，首选
cue.ir_mul。use_fallback (bool, optional) – 如果为 None（默认），则在可用时将使用 CUDA 内核。如果为 False，将使用 CUDA 内核，如果不可用，则会引发异常。如果为 True，则无论 CUDA 内核是否可用，都将使用 PyTorch 后备方法。
前向传播