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)