Hi,
I have 3D and 4D tensors as follows:
A = [16,16,13] and B = [16,16,13,49]
And I would like to get tensor [16,16,49] without performing loop iterations.
Currently I’m doing:
for in_i in range(16):
for in_j in range(16):
results[in_i,in_j,:] = torch.matmul(A[in_i, in_j , :], B[in_i, in_j, :, :])
Any idea?
Thanks in advance!