mindscience.e3nn.o3.FullyConnectedTensorProduct

class mindscience.e3nn.o3.FullyConnectedTensorProduct(irreps_in1, irreps_in2, irreps_out, ncon_dtype=mindspore.float32, **kwargs)[源代码]

全连接加权张量积。所有满足 \(|l_1 - l_2| \leq l_{out} \leq l_1 + l_2\) 的路径都将被考虑。等效于 TensorProduct 的 instructions='connect'。详细信息请参见 mindscience.e3nn.o3.TensorProduct 。

参数：

irreps_in1 (Union[str, Irrep, Irreps]) - 第一个输入的 Irreps。
irreps_in2 (Union[str, Irrep, Irreps]) - 第二个输入的 Irreps。
irreps_out (Union[str, Irrep, Irreps]) - 输出的 Irreps。
ncon_dtype (mindspore.dtype, 可选) - ncon 计算模块输入张量的类型。默认值：mindspore.float32。

样例：

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