Hi there,
Say if I got two tensors like [[1,1],[1,1]] and [[2,2],[2,2]], how could I interleave them along n_w or n_h dimension to get [[1,2,1,2],[1,2,1,2]] or [[1,1],[2,2],[1,1],[2,2]]? In TensorFlow I could achieve such goal using tf.reshape after tf.stack, but transferring to PyTorch using view after torch.stack does not return what I want.
Thanks.