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