Splitting an input tensor

Didn’t find “nn.SplitTable” that is available in Torch Lua. Doing this in order to make it goes in a split to split convolution process. What is the equivalent Plz?
line : seq:add(nn.SplitTable(d+1, 3)).

Have you tried the torch.split (https://pytorch.org/docs/stable/torch.html#torch.split) or torch.chunk (https://pytorch.org/docs/stable/torch.html#torch.chunk) function?

1 Like

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