TensorFlow:tf.reverse函數(shù)

2018-12-26 10:53 更新

tf.reverse 函數(shù)

reverse(
    tensor,
    axis,
    name=None
)

定義在:tensorflow/python/ops/array_ops.py.

參見(jiàn)指南:張量變換>分割和連接

反轉(zhuǎn)張量的特定維度.

注意tf.reverse現(xiàn)在已經(jīng)改變了為1.0準(zhǔn)備的行為.tf.reverse_v2目前是一個(gè)別名,將在 tf 1.0 之前棄用.

給定一個(gè)tensor和一個(gè)int32類(lèi)型的張量axis,以表示要反轉(zhuǎn)的tensor的一組維度.此操作反轉(zhuǎn)每個(gè)維度 i ,其中存在 j,例如:axis[j] == i.

tensor可以有多達(dá)8個(gè)維度.指定的維數(shù)axis可以是0或更多的條目.如果多次指定索引,則會(huì)引發(fā)InvalidArgument錯(cuò)誤.

例如:

# tensor 't' is [[[[ 0,  1,  2,  3],
#                  [ 4,  5,  6,  7],
#                  [ 8,  9, 10, 11]],
#                 [[12, 13, 14, 15],
#                  [16, 17, 18, 19],
#                  [20, 21, 22, 23]]]]
# tensor 't' shape is [1, 2, 3, 4]

# 'dims' is [3] or 'dims' is -1
reverse(t, dims) ==> [[[[ 3,  2,  1,  0],
                        [ 7,  6,  5,  4],
                        [ 11, 10, 9, 8]],
                       [[15, 14, 13, 12],
                        [19, 18, 17, 16],
                        [23, 22, 21, 20]]]]

# 'dims' is '[1]' (or 'dims' is '[-3]')
reverse(t, dims) ==> [[[[12, 13, 14, 15],
                        [16, 17, 18, 19],
                        [20, 21, 22, 23]
                       [[ 0,  1,  2,  3],
                        [ 4,  5,  6,  7],
                        [ 8,  9, 10, 11]]]]

# 'dims' is '[2]' (or 'dims' is '[-2]')
reverse(t, dims) ==> [[[[8, 9, 10, 11],
                        [4, 5, 6, 7],
                        [0, 1, 2, 3]]
                       [[20, 21, 22, 23],
                        [16, 17, 18, 19],
                        [12, 13, 14, 15]]]]

參數(shù):

  • tensor:一個(gè)Tensor.必須是下列類(lèi)型之一:uint8,int8,uint16,int16,int32,int64,bool,half,float32,float64,complex64,complex128,string.可以達(dá)到8-D.
  • axis:一個(gè) Tensor.必須是以下類(lèi)型之一:int32,int64;1-d.要反轉(zhuǎn)的維度的指數(shù).必須在范圍[-rank(tensor), rank(tensor))內(nèi).
  • name:操作的名稱(可選).

返回:

該函數(shù)將返回一個(gè) Tensor.與tensor具有相同的類(lèi)型和形狀.

以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)