2D times 2D equals a 3d pytorch tensor

Question

Given two 2-D pytorch tensors:

A=torch.FloatTensor([[1,2],[3,4]]) B=torch.FloatTensor([[0,0],[1,1],[2,2])

Is there any efficient way to calculate a tensor (2,2