mindchemistry.e3.o3.so3_generators

查看源文件
mindchemistry.e3.o3.so3_generators(l, dtype=float32)[源代码]

计算so(3)李代数生成元。

参数:
  • l (int) - 生成维度。

  • dtype (dtype) - {float32, float64}生成器的数据类型。默认值: float32

返回:
  • output (Tensor) - 张量,计算so(3)李代数生成元结果,类型为dtype。

异常:
  • TypeError - 如果 j 不是整型。

  • ValueError - 如果矩阵数据不一致。

支持平台:

Ascend

样例:

>>> from mindchemistry.e3.o3 import so3_generators
>>> m = so3_generators(1)
>>> print(m)
[[[ 0.          0.          0.        ]
[ 0.          0.         -0.99999994]
[ 0.          0.99999994  0.        ]]
[[ 0.          0.          0.99999994]
[ 0.          0.          0.        ]
[-0.99999994  0.          0.        ]]
[[ 0.         -0.99999994  0.        ]
[ 0.99999994  0.          0.        ]
[ 0.          0.          0.        ]]]