mindscience.e3nn.o3.so3_generators

mindscience.e3nn.o3.so3_generators(l, dtype=mindspore.float32)[源代码]

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

参数:
  • l (int) - 生成元的阶数。

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

返回:

Tensor,so(3) 生成元,数据类型为 mindspore.dtype

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

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

样例:

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