With two tensors
a = torch.ones([256, 512, 32])
b = torch.ones([32, 2])

what is the most efficient way to broadcast b onto every associated entry in a, producing a result with shape
[256, 512, 32, 2] ?
Is there an inplace variant maybe? I want to avoid making copies as much as possible…