mindscience.e3nn.o3.FullyConnectedTensorProduct

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

Fully-connected weighted tensor product. All the possible path allowed by \(|l_1 - l_2| \leq l_{out} \leq l_1 + l_2\) are made. Equivalent to TensorProduct with instructions='connect'. For details, see mindscience.e3nn.o3.TensorProduct.

Parameters

irreps_in1 (Union[str, Irrep, Irreps]) – Irreps for the first input.
irreps_in2 (Union[str, Irrep, Irreps]) – Irreps for the second input.
irreps_out (Union[str, Irrep, Irreps]) – Irreps for the output.
ncon_dtype (mindspore.dtype, optional) – The type of input tensors of ncon computation module. Default: mindspore.float32.

Examples

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