1. 程式人生 > >tensorflow.reshap(tensor,shape,name)的使用說明

tensorflow.reshap(tensor,shape,name)的使用說明

light 自動 推出 存在 pes highlight brush sha 我們

tensorflow as tf

tf.reshape(tensor, shape, name=None)
reshape作用是將tensor變換為指定shape的形式。
其中shape為一個列表形式,特殊的一點是列表中可以存在-1。-1代表的含義是不用我們自己指定這一維的大小,函數會自動計算(根據已給定的維度,自動推出-1指定的維度),但列表中只能存在一個-1。(當然如果存在多個-1,就是一個存在多解的方程了)

# tensor ‘t‘ is [1, 2, 3, 4, 5, 6, 7, 8, 9]
# tensor ‘t‘ has shape [9]
reshape(t, [3, 3]) ==> [[1, 2, 3],
                        [4, 5, 6],
                        [7, 8, 9]]

# tensor ‘t‘ is [[[1, 1], [2, 2]],
#                [[3, 3], [4, 4]]]
# tensor ‘t‘ has shape [2, 2, 2]
reshape(t, [2, 4]) ==> [[1, 1, 2, 2],
                        [3, 3, 4, 4]]

# tensor ‘t‘ is [[[1, 1, 1],
#                 [2, 2, 2]],
#                [[3, 3, 3],
#                 [4, 4, 4]],
#                [[5, 5, 5],
#                 [6, 6, 6]]]
# tensor ‘t‘ has shape [3, 2, 3]
# pass ‘[-1]‘ to flatten ‘t‘
reshape(t, [-1]) ==> [1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6]

# -1 can also be used to infer the shape

# -1 is inferred to be 9:
reshape(t, [2, -1]) ==> [[1, 1, 1, 2, 2, 2, 3, 3, 3],
                         [4, 4, 4, 5, 5, 5, 6, 6, 6]]
# -1 is inferred to be 2:
reshape(t, [-1, 9]) ==> [[1, 1, 1, 2, 2, 2, 3, 3, 3],
                         [4, 4, 4, 5, 5, 5, 6, 6, 6]]
# -1 is inferred to be 3:
reshape(t, [ 2, -1, 3]) ==> [[[1, 1, 1],
                              [2, 2, 2],
                              [3, 3, 3]],
                             [[4, 4, 4],
                              [5, 5, 5],
                              [6, 6, 6]]]

# tensor ‘t‘ is [7]
# shape `[]` reshapes to a scalar
reshape(t, []) ==> 7

  

tensorflow.reshap(tensor,shape,name)的使用說明