如何在PyTorch中将两个张量轴相乘？

Question

我有两个形状的3D张量：

a = torch.full([1495, 110247, 1], 0.5)
b = torch.full([1495, 110247, 2], 1)

我想将它们相乘，以使前两个维保持不变，第三个维是a和b的第三个维的标量积。