a.shape = [1, 3, 80, 64, 2]
b.shape = [1, 3, 80, 64, 85]
b[..., 0:2] = a
there is always wrong with … in my environment, so is there some way to replace …? tkx!
What kind of error are you getting when using the ellipsis?
Instead of using the three dots you could also replace it with the Ellipsis keyword, but I guess this would yield the same error you are seeing.
Note that your code snippet works fine for me.
LOL
b = torch.cat([a, b[:,:,:,:,2:]], -1)