Adding documentation for every function in layers.py (#5529)

* Adding operator assignment

* Adding documentation to layers.py

* Removing file from another PR

Author: kavyasrinet