mindscience.e3nn.o3.ElementwiseTensorProduct

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

元素级连接张量积。

等效于 TensorProduct 的 instructions='element'。详细信息请参见 mindscience.e3nn.o3.TensorProduct。

参数：

irreps_in1 (Union[str, Irrep, Irreps]) - 第一个输入的 Irreps。
irreps_in2 (Union[str, Irrep, Irreps]) - 第二个输入的 Irreps。
filter_ir_out (Union[str, Irrep, Irreps, None], 可选) - 过滤器，仅选择特定的输出 Irrep。默认值：None。
ncon_dtype (mindspore.dtype, 可选) - ncon 计算模块输入张量的类型。默认值：mindspore.float32。

样例：

>>> from mindscience.e3nn.o3 import ElementwiseTensorProduct
>>> ElementwiseTensorProduct('2x2e+4x1o', '3x1e+3x0o')
TensorProduct [element] (2x2e+1x1o+3x1o x 2x1e+1x1e+3x0o -> 2x1e+2x2e+2x3e+1x0o+1x1o+1x2o+3x1e)