Solved by bhushans23 in post #2 
              
              
                You can use repeat to repeat the tensor n times 
But, will need to convert the tensor to batch format first. 

Reshape to 1 x size x size
Use repeat and mention batchsize x 1 x 1

Following is an example 
x = torch.eye(3)
x = x.reshape((1, 3, 3))
y = x.repeat(5, 1, 1)
print(x.shape)
print(y.shape)

…