Add support for `torch.nn.functional.normalize`

[MiWeiss]

There is currently no support for torch.nn.functional.normalize.

See also this comment: #707 (comment)

I'll have a go trying to open a PR with this functionality, but can't make any promises :-)