Columns 2D tenson times rows 2D tensor equals a 3d pytorch tensor

Question

Given 2 tensors 2-D in PyTorch A (a X m) and B(m X b), is there any efficient way to obtain a tensor C (m X a X b), where C[i,:,:] = A[:,i] @ B[i,:]?

Here I will give