mindscience.e3nn.o3.Linear

class mindscience.e3nn.o3.Linear(irreps_in, irreps_out, ncon_dtype=mindspore.float32, **kwargs)[源代码]

线性等变操作。 等效于 TensorProductinstructions='linear'。详细信息请参见 mindscience.e3nn.o3.TensorProduct

参数:
  • irreps_in (Union[str, Irrep, Irreps]) - 输入的 Irreps。

  • irreps_out (Union[str, Irrep, Irreps]) - 输出的 Irreps。

  • ncon_dtype (mindspore.dtype, 可选) - 用于ncon的数据类型。默认值:mindspore.float32

样例:

>>> from mindscience.e3nn.o3 import Linear
>>> Linear('2x2e+3x1o+3x0e', '3x2e+5x1o+2x0e')
TensorProduct [linear] (2x2e+3x1o+3x0e x 1x0e -> 3x2e+5x1o+2x0e)