Splitting an input tensor

I found this sheet: Directory for Table Layers in PyTorch
Thnx Caruso