layer_dense_torch function

Function to define a torch layer similar to a tf dense layer