具体代码
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))