mindchemistry.e3.o3.FullTensorProduct

查看源文件
class mindchemistry.e3.o3.FullTensorProduct(irreps_in1, irreps_in2, filter_ir_out=None, ncon_dtype=float32, **kwargs)[源代码]

完全张量积。

等效于 TensorProduct 的 instructions='full'。详细信息请参见 mindchemistry.e3.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。

  • irrep_norm (str) - {'component', 'norm'}，输入和输出表示的假定归一化。默认值：component。

  • path_norm (str) - {'element', 'path'}，路径权重的归一化方法。默认值：element。

  • weight_init (str) - {'zeros', 'ones', 'truncatedNormal', 'normal', 'uniform', 'he_uniform', 'he_normal', 'xavier_uniform'}，权重的初始化方法。默认值：normal。

  • ncon_dtype (mindspore.dtype) - ncon 计算模块输入张量的类型。默认值：mindspore.float32。

支持平台：

Ascend

样例：

>>> from mindchemistry.e3.o3 import FullTensorProduct
>>> FullTensorProduct('2x1o+4x0o', '1x1o+3x0e')
TensorProduct [full] (2x1o+4x0o x 1x1o+3x0e -> 2x0e+12x0o+6x1o+2x1e+4x1e+2x2e)