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_xinput_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]