mindsponge.common.initial_affine

mindsponge.common.initial_affine(num_residues, use_numpy=False)[源代码]

初始化仿射变换后的四元数，旋转矩阵，平移向量。

参数：

num_residues (int) - 氨基酸残基数量。
use_numpy (bool) - 是否使用numpy计算，默认值：False。

返回：

返回初始化后仿射变换结果 - 四元数 (tensor)，shape为 \((N_{res}, 4)\) 。 - 旋转矩阵 (tuple) \((xx, xy, xz, yx, yy, yz, zx, zy, zz)\) ，每个元素shape为 \((N_{res}, )\) 。 - 平移向量 (tuple) \((x, y, z)\) ，每个元素shape为 \((N_{res}, )\) 。

支持平台：

Ascend GPU

样例：

>>> import numpy as np
>>> import mindspore as ms
>>> from mindspore import Tensor
>>> from mindsponge.common.geometry import initial_affine
>>> output = initial_affine(256)
>>> print(len(output), output[0].shape, len(output[1]), len(output[1][0]), len(output[2]), len(output[2][0]))
>>> print(output[0])
>>> print(output[1])
>>> print(output[2])
3, (1, 4), 9, 1, 3, 1
[[1.00000000e+00, 0.00000000e+00, 0.00000000e+00, 0.00000000e+00]]
(1, 0, 0, 0, 1, 0, 0, 0, 1)
([0.00000000e+00], [0.00000000e+00], [0.00000000e+00])