mindspore.ops.RightShift
- class mindspore.ops.RightShift[源代码]
将Tensor input_x 的每个元素右移 Tensor input_y 中对应位数。输入为两个Tensor,数据类型需要保持一致,他们之间的shape可以广播。
\[\begin{aligned} &out_{i} =x_{i} >> y_{i} \end{aligned}\]警告
这是一个实验性API,后续可能修改或删除。
- 输入:
input_x (Tensor) - 被操作Tensor,将被逐元素位右移 input_y 位。支持所有int和uint类型。
input_y (Tensor) - 右移位数。数据类型必须和 input_x 一致。
- 输出:
output (Tensor) - 输出Tensor,据类型和 input_x 一致。
- 异常:
TypeError - input_x 或者 input_y 不是Tensor。
TypeError - input_x 和 input_y 不能发生广播。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import numpy as np >>> from mindspore import Tensor, ops >>> input_x = Tensor(np.array([1, 2, 3]).astype(np.uint8)) >>> input_y = Tensor(np.array([1, 1, 1]).astype(np.uint8)) >>> output = ops.RightShift()(input_x, input_y) >>> print(output) [0 1 1]