Okay this worked similar to keras.backend.repeat_elements()
import torch

A = torch.randn([5,3,224,224])

print(A.shape)

B = torch.repeat_interleave(A,4,dim=1)

print(B.shape)  ## (5,12,224,224)

Probably I was missing something
Thanks !!