Solved by LeviViana in post #2 
              
              
                This way: 
import torch

n = 3
a = torch.arange(n).unsqueeze(1) # [n, 1]
b = torch.ones(4).view(2, 2) # [2, 2]

a[:, :, None] * b # [n, 2, 2]