tf.reverse_v2函數(shù):反轉(zhuǎn)張量的特定維度

2018-12-28 10:31 更新

tf.reverse_v2 函數(shù)

reverse_v2(
    tensor,
    axis,
    name=None
)

參見指南:張量變換>分割和連接

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

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

給定一個tensor和一個int32型張量axis,表示一組要反轉(zhuǎn)的tensor的維度.此操作反轉(zhuǎn)存在j s.t. axis[j] == i的每個維度i.

tensor最多可以有8個維度.axis指定的維數(shù)可以是0或更多的條目.如果指定的索引不止一次,則會引發(fā)InvalidArgument錯誤.

例如:

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

返回:

該函數(shù)將返回一個Tensor,它與tensor有相同的類型和形狀.

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

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號