In this video, we will learn how to create custom layers on TensorFlow using Keras API. For this tutorial, we are going to create a custom Dense layer by extending the tf.keras.layers.Layer class. For a fair comparison, first, we are going to build a simple MNIST classifier using regular Dense layer from the Keras API. After that, we are going to replace the regular Dense layer with the CustomDense layer.

CODE: https://github.com/nikhilroxtomar/Custom-Layer-in-TensorFlow-using-Keras-API

Subscribe: https://www.youtube.com/channel/UClkqp31PHke-f8b8mjiiY-Q

#tensorflow #keras