1. 程式人生 > >TensorFlow教程——nest.flatten()函式解析

TensorFlow教程——nest.flatten()函式解析

函式作用:將巢狀結構壓平,返回Python的list。

例子一,巢狀列表:

from tensorflow.python.util import nest

input = [['a', 'b', 'c'],
        ['d', 'e', 'f'],
        ['1', '2', '3']]

result = nest.flatten(input)

Out:
<class 'list'>: ['a', 'b', 'c', 'd', 'e', 'f', '1', '2', '3']

例子二,字典型別:

input = {'a': 1, 'b': 2
, 'c': {'p': 3, 'q': 4}} result = nest.flatten(input) Out: <class 'list'>: [1, 2, 3, 4]

注意:
1)Tensor物件、和NumPy的ndarray物件,都看做是一個元素,不會被壓平。

input = {'a': np.ones(shape=(2, 3)), 'b': np.zeros(shape=(3, 2))}

result = nest.flatten(input)

Out:
<class 'list'> [ndarray, ndarray]

2)如果輸入不是巢狀結構,只有一個元素的話,則返回僅包含這一個元素的list。

from tensorflow.python.util import nest

x = tf.constant([[1,2],[3,4]])
y = nest.flatten(x)

type(y) # list
len(y) # 1
y[0] # <tf.Tensor 'Const_4:0' shape=(2, 2) dtype=int32>

3)注意和tf.contrib.layers.flatten()的區別。