python - 如何通过广播对具有不同维度的 2 个张量进行元素分析？

Question

我有一个inputs用 a调用的张size量torch.Size([20, 1, 161, 199])和另一个mask用torch.Size([20, 1, 199]). 我想将它们相乘。

我试过：

masked_inputs = inputs * mask[..., None]

但得到一个错误：

RuntimeError: The size of tensor a (161) must match the size of tensor b (199) at non-singleton dimension 2

我不太确定该怎么办？

标签： pythonpytorchtensor

Answer 1

这做到了：

masked_inputs = inputs * mask.unsqueeze(2)