mindscience.e3nn.o3.so3_generators

mindscience.e3nn.o3.so3_generators(l, dtype=mindspore.float32)[source]

Compute the so(3) Lie algebra generators.

Parameters
  • l (int) – degree of generators.

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

Returns

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

Raises

Examples

>>> from mindscience.e3nn.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.        ]]]