TransposeIrrepsLayout

class cuequivariance_torch.TransposeIrrepsLayout(

irreps: Irreps,

*,

source: IrrepsLayout,

target: IrrepsLayout,

device: device | None = None,

use_fallback: bool | None = None,

)

转置张量的 irreps 布局。

参数:

• irreps (Irreps) – 张量的 irreps。

• source (IrrepsLayout) – 源布局。

• target (IrrepsLayout) – 目标布局。

前向传播

forward(x: Tensor) → Tensor

执行转置。

参数:

• x (torch.Tensor) – 输入张量。

• use_fallback (bool, optional) – 如果 None (默认)，如果可用将使用 CUDA 内核。如果 False，将使用 CUDA 内核，如果不可用则会引发异常。如果 True，无论 CUDA 内核是否可用，都将使用 PyTorch 回退方法。

返回值:

转置后的张量。

返回类型:

torch.Tensor