Adding capability to choose device other than cpu and fixing/generalize#96

Open

arnabdas2019ovgu wants to merge 1 commit intoutkuozbulak:masterfrom

arnabdas2019ovgu:generalize

Commits on Dec 3, 2020

Adding capability to chhose devce other than cpu and fixing/generalising methods for layer specific activations by introducing new hooks.
arnabdas2019ovgu
committed