sponge.function.gather_value

sponge.function.gather_value(tensor: Tensor, index: Tensor)

根据指标从张量的最后一根轴收集值。

参数：

• tensor (Tensor) - 输入张量，shape为 \((B, X)\) 。其中 \(B\) 为batch size， \(X\) 为输入张量第二维的任意大小。

• index (Tensor) - 索引，shape为 \((B, ...,)\) 。

返回：

Tensor。取出的值，shape为 \((B, ...,)\) 。

支持平台：

Ascend GPU