TensorFlow函數(shù)：tf.manip.roll

tf.manip.roll函數(shù)

tf.manip.roll(
    input,
    shift,
    axis
)

定義在：tensorflow/python/ops/manip_ops.py.

沿軸線滾動(dòng)張量的元素.

沿著axis的維度,通過shift的偏移,元素正向移動(dòng)(朝向更大的指數(shù)).

負(fù)值shift會(huì)使元素向相反方向移動(dòng).滾動(dòng)通過最后位置的元素將環(huán)繞到第一個(gè)位置,反之亦然.可以指定沿多個(gè)軸的多個(gè)移位.

例如：

# 't' is [0, 1, 2, 3, 4]
roll(t, shift=2, axis=0) ==> [3, 4, 0, 1, 2]

# shifting along multiple dimensions
# 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]
roll(t, shift=[1, -2], axis=[0, 1]) ==> [[7, 8, 9, 5, 6], [2, 3, 4, 0, 1]]

# shifting along the same axis multiple times
# 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]
roll(t, shift=[2, -3], axis=[1, 1]) ==> [[1, 2, 3, 4, 0], [6, 7, 8, 9, 5]]

參數(shù)：

• input：一個(gè)Tensor.
• shift：一個(gè)Tensor；必須是以下類型之一：int32,int64.維度必須為0-D或1-D；shift[i]指定元素沿由axis[i]指定的維度正向(朝向較大索引)移位的位置數(shù).負(fù)向移動(dòng)將使元素向相反方向滾動(dòng).
• axis：一個(gè)Tensor；必須是以下類型之一：int32,int64.維度必須為0-D或1-D；axis[i]指定移動(dòng)shift[i]應(yīng)發(fā)生的維度.如果多次引用同一軸,則該軸的總偏移將是屬于該軸的所有偏移的總和.
• name：操作的名稱(可選).

返回：

該函數(shù)返回一個(gè)Tensor,它與input有相同的類型.