import tensorflow as tf
dense = tf.keras.layers.Dense(
7, # 输出最后1维的维度
activation="relu" # 激活函数
)
print(dense)
<tensorflow.python.keras.layers.core.Dense object at 0x0000026B5A025820>
input的维度必须是大于1维
inputs = tf.random.normal([
32, # 第1个维度是样本数
10,
8,
])
print(inputs.shape)
(32, 10, 8)
output的维度会继承上层的维度
output = dense(inputs)
print(output.shape)
(32, 10, 7)