tf.keras.layers.Flatten

展平层

keras.layers.Flatten(data_format=None)

将输入张量展平，不影响批量大小。

data_format 表示输入张量的维度，默认为[batch, height, width, channel]

示例

from tensorflow.keras.layers import Flatten
import tensorflow as tf
import numpy as np# 定义一个展平层
flatten = Flatten()# 生成一个维度为[64, 720, 720, 3]的矩阵
x = np.random.random((64, 720, 720, 3))# 转成tensor类型,第一个维度64表示batch
# numpy中的数据类型和tensorflow中的数据类型完全兼容,所以这一步可以省略
x = tf.convert_to_tensor(x)
print(x.shape) # [64, 720, 720, 3]# 进行展平
y = flatten(x)
print(y.shape) # [64, 1555200]

tf.keras.layers.Flatten相关推荐

tf.keras.layers.Flatten() 示例
import tensorflow as tf flatten = tf.keras.layers.Flatten() print(flatten) <tensorflow.python.ker ...
tf.keras.layers.Flatten该网络中的第一层tf.keras.layers.Flatten将图像的格式从2d阵列（28乘28像素）转换为28 * 28 = 784像素的1d阵列。可以
https://blog.csdn.net/qq_20989105/article/details/82760815
全连接层tf.keras.layers.Dense()介绍
函数原型 tf.keras.layers.Dense(units, # 正整数,输出空间的维数activation=None, # 激活函数,不指定则没有use_bias=True, # 布尔值,是否 ...
深入浅出TensorFlow2函数——tf.keras.layers.Dense
分类目录:<深入浅出TensorFlow2函数>总目录 tf.keras.layers.Dense实现操作:output = activation(dot(input, kernel) + ...
tensorflow keras 上采样（放大图片） tf.keras.layers.UpSampling2D 示例
input_shape = (4, 28, 28, 3) # 样本数:4,图片大小:28 * 28,通道:3 inputs = tf.random.normal(input_shape) print( ...
tf.keras.layers.MaxPool2D 最大池化层示例
tf.keras.layers.MaxPool2D 最大池化层示例 import tensorflow as tf import numpy as np inputs = np.random.ran ...
tf.keras.layers.Conv1D 一维卷积示例
tf.keras.layers.Conv1D 一维卷积示例 import tensorflow as tf from tensorflow import keras import numpy as ...
tf.keras.layers.MaxPool2D 示例池化层
tf.keras.layers.MaxPool2D 示例池化层 import tensorflow as tf import numpy as np inputs = np.random.randi ...
tf.keras.layers.Resizing 示例改变维度的层
tf.keras.layers.Resizing 示例改变维度的层 import tensorflow as tf # 创建一个输入的张量 inputs = tf.random.normal([ 2 ...

tf.keras.layers.Flatten

展平层

示例

tf.keras.layers.Flatten相关推荐

最新文章

热门文章