mindsponge.common.vecs_from_tensor
- mindsponge.common.vecs_from_tensor(inputs)[源代码]
将输入表示位置信息的tensor在最后一根轴拆分,化为向量, vecs_to_tensor 的逆操作。
- 参数:
inputs (Tensor) - 原子位置信息,shape为 \((..., 3)\)。
- 返回:
返回带有三个tensor的tuple \((x, y, z)\),分别包含x, y, z坐标信息。
- 支持平台:
Ascend
GPU
样例:
>>> import numpy as np >>> import mindspore as ms >>> from mindspore import Tensor >>> from mindsponge.common.geometry import vecs_from_tensor >>> input_0 = Tensor(np.ones((4, 256, 3)), ms.float32) >>> output = vecs_from_tensor(input_0) >>> print(len(output), output[0].shape) 3, (4,256)