I have to perform the operation:

I have two tensors that I want to multiply in this way, `A`

and `B`

:

```
A = torch.randn(B,C,j,i)
B = torch.randn(B,C,j,k)
C = AxB
C.shape = torch.Size(B,C,j,i,k)
```

Where I have assumed the shape of the output tensor based on the indices of the equation.

What is the function that I should use to produce the desired result?