Python tf.raw_ops.RightShift用法及代碼示例

Elementwise 計算 x 和 y 的按位 right-shift。

用法

tf.raw_ops.RightShift(
    x, y, name=None
)

對無符號整數類型執行邏輯移位，對有符號整數類型執行算術移位。

如果y 為負數，或大於或等於x 的寬度(以位為單位)，則結果由實現定義。

例子：

import tensorflow as tf
from tensorflow.python.ops import bitwise_ops
import numpy as np
dtype_list = [tf.int8, tf.int16, tf.int32, tf.int64]

for dtype in dtype_list:
  lhs = tf.constant([-1, -5, -3, -14], dtype=dtype)
  rhs = tf.constant([5, 0, 7, 11], dtype=dtype)

  right_shift_result = bitwise_ops.right_shift(lhs, rhs)

  print(right_shift_result)

# This will print:
# tf.Tensor([-1 -5 -1 -1], shape=(4,), dtype=int8)
# tf.Tensor([-1 -5 -1 -1], shape=(4,), dtype=int16)
# tf.Tensor([-1 -5 -1 -1], shape=(4,), dtype=int32)
# tf.Tensor([-1 -5 -1 -1], shape=(4,), dtype=int64)

lhs = np.array([-2, 64, 101, 32], dtype=np.int8)
rhs = np.array([-1, -5, -3, -14], dtype=np.int8)
bitwise_ops.right_shift(lhs, rhs)
# <tf.Tensor:shape=(4,), dtype=int8, numpy=array([ -2,  64, 101,  32], dtype=int8)>