Torch.repeat_interleave on dictionary?

[ptrblck]

Yes, torch.repeat_interleave expects a tensor as its input.