3
Tensor("ones:0", shape=(1, 2), dtype=float32)
Number of weights after calling the model: 6
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
layer1 (Dense) multiple 6
_________________________________________________________________
layer2 (Dense) multiple 9
_________________________________________________________________
layer3 (Dense) multiple 16
=================================================================
Total params: 31
Trainable params: 31
Non-trainable params: 0
_________________________________________________________________
WARNING:tensorflow:From /usr/local/lib/python3.7/dist-packages/tensorflow/python/ops/resource_variable_ops.py:435: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version.
Instructions for updating:
Colocations handled automatically by placer.