If I have a tensor like x = torch.rand((3,4,8)) and I would like to slice ‘x’ in order to fit into y = torch.rand((2,3,4,4)).
I am able to do slicing in 2D using torch.narrow or torch.select. But, I am really confused in the 3D.
I really need to know this because I want to split up a bunch of patches.