megengine.functional.right_shift

right_shift(x, y)[源代码]

元素右移。

根据输入张量 \(x_i\) 的相应元素 \(y_i\),将输入张量 \(y\) 的每个元素的位向右移动。

注解

>> 运算符可以作为在 Tensors 上进行 `F. right_shift’ 操作的简写。

注解

这个操作必须是一个算术移位(即符号传播),因此等同于向下取整除以 2 的幂。

参数
  • x – 第一个输入张量。应该有一个整数数据类型。

  • y – 第二个输入张量。必须与 \(x\) 兼容(参见 广播机制与规则 )。应该有一个整数数据类型。每个元素必须大于或等于 0

返回

一个 Tensor,包含输入 Tensor 逐元素向右移的结果。该 Tensor 的类型由 类型提升规则 决定。

实际案例

>>> F.right_shift([2, 4, 8], 1)
Tensor([1 2 4], dtype=int32, device=xpux:0)

逐个元素的左移:

>>> x = Tensor([2, 8, 24])
>>> y = Tensor([1, 2, 3])
>>> F.right_shift(x, y)
Tensor([1 2 3], dtype=int32, device=xpux:0)

广播:

>>> F.right_shift([10, 20, 40], 2)
Tensor([ 2  5 10], dtype=int32, device=xpux:0)