Tesorflow2.0 全连接层（Dense）

具体代码

import sys

import matplotlib as mpl
import numpy as np
import pandas as pd
import sklearn
import tensorflow as tf
from tensorflow import keras

print(tf.__version__)
print(sys.version_info)

for module in mpl, np, pd, sklearn, keras, tf:
    print(module.__name__, module.__version__)

# dense_layer = tf.keras.layers.Dense(100)
# 全连接层 神经网络个数 100个 输入形状 (b, 5)
dense_layer = tf.keras.layers.Dense(100, input_shape=(None, 5))
# 输入
input_data = tf.random.normal((10, 5))
out_put = dense_layer(input_data)  # 输出 (10, 100)

#  x @ w + b = y
#  input  @ kernel(variables) + bias = output
# [10, 5] @ [5, 100] + [100] = [10, 100]
print(dense_layer.variables)  # 变量 kernel(variables) + bias
print(dense_layer.trainable_variables)  # 可训练变量 kernel(variables) + bias
# print(help(dense_layer))