tensorflow之神经网络层:Flatten,flatten

版权声明:微信公众号:数据挖掘与机器学习进阶之路。本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u013230189/article/details/82775927

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)

没有更多推荐了,返回首页