tensorflow之神经网络层:Flatten,flatten

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

)

参数:

  1. inputs:输入的张量
  2. name:层的名称

x=tf.placeholder(shape=(None,4,4),dtype='float32')

y=tf.layers.flatten(x)

print(y)

你可能感兴趣的:(深度学习,tensorflow)