mindchemistry.e3.o3.so3_generators

View Source On Gitee
mindchemistry.e3.o3.so3_generators(l, dtype=float32)[source]

Compute the so(3) Lie algebra generators.

Parameters
  • l (int) – degree of generators.

  • dtype (dtype) – {float32, float64} data type of generators. Default: float32.

Returns

Tensor, so(3) generators with the dtype is dtype.

Raises
Supported Platforms:

Ascend

Examples

>>> 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.        ]]]