Hetero support

[doGregor]

• fixed representation issue in documentation for heterogeneous data structures
• adapted train/test split function to heterogeneous data structures
• tests for train/test split function (signals and batches)

[benedekrozemberczki]

Thank you!