torch.split & torch.chunk

The PR adds support for torch.split and torch.chunk. Includes unit tests and documentation.