mindsponge.common.vecs_to_tensor

mindsponge.common.vecs_to_tensor(v)[源代码]

将向量转化为最后一维度为3的tensor,和vecs_from_tensor操作相反。

参数:
  • v (tuple) - 带有3个tensor,分别表示位置坐标中的x, y, z。

返回:

tensor,返回三个tensor在最后一根轴合并后结果,shape为 \((..., 3)\)

支持平台:

Ascend GPU

样例:

>>> import numpy as np
>>> import mindspore as ms
>>> from mindspore import Tensor
>>> from mindsponge.common.geometry import vecs_to_tensor
>>> input_0 = Tensor(np.ones((4, 256)), ms.float32)
>>> input_1 = Tensor(np.ones((4, 256)), ms.float32)
>>> input_2 = Tensor(np.ones((4, 256)), ms.float32)
>>> inputs = (input_0, input_1, input_2)
>>> output = vecs_to_tensor(inputs)
>>> print(output.shape)
(4, 256, 3)