1.tf.layers.Flatten
Class Flatten:在保留第0轴的情况下对输入的张量进行Flatten(扁平化)
代码示例:
x=tf.placeholder(shape=(None,4,4),dtype='float32') y=tf.layers.Flatten()(x) print(y) |
输出:
Tensor("flatten/Reshape:0", shape=(?, 16), dtype=float32)
2.tf.layers.flatten
同Flatten类。
tf.layers.flatten( inputs, name=None ) |
参数:
- inputs:输入的张量
- name:层的名称
x=tf.placeholder(shape=(None,4,4),dtype='float32') y=tf.layers.flatten(x) print(y) |