mindsponge.common.vecs_to_tensor
- mindsponge.common.vecs_to_tensor(v)[源代码]
将向量转化为最后一维度为3的tensor,和vecs_from_tensor操作相反。
- 参数:
v (tuple) - 带有3个tensor的tuple \((x, y, z)\) ,分别表示位置坐标中的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)