import tensorflow as tf
from tensorflow import keras

x = tf.random.normal([4, 784])
print(x)

net = keras.layers.Dense(10)
print(net)
net.build((4, 784))
print(net)

print(net(x).shape)
print(net.kernel.shape)
print(net.bias.shape)