mindspore.ops.unsorted_segment_prod

mindspore.ops.unsorted_segment_prod(x, segment_ids, num_segments)

分段计算输入tensor的乘积。

unsorted_segment_prod的计算过程如下图所示：

说明

• 如果 segment_ids 中不存在segment_id i ，则将使用1填充输出 output[i] 。

• segment_ids 必须是一个非负tensor。

参数：

• x (Tensor) - 输入tensor。

• segment_ids (Tensor) - 标识每个元素所属段。

• num_segments (Union[int, Tensor], 可选) - 分段数量，可以为int或零维的tensor。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[1, 2, 3], [4, 5, 6], [4, 2, 1]])
>>> segment_ids = mindspore.tensor([0, 1, 0])
>>> num_segments = 2
>>> mindspore.ops.unsorted_segment_prod(x, segment_ids, num_segments)
Tensor(shape=[2, 3], dtype=Int64, value=
[[4, 4, 3],
 [4, 5, 6]])