nn_unflatten function

Unflattens a tensor dim expanding it to a desired shape. For use with [nn_sequential .