如何根据另一个张量的值更改PyTorch张量的一部分？

Question

这个问题可能不清楚，因此请在评论中进行澄清，我将继续进行讨论。

我的以下张量具有以下形状：

mask.size() == torch.Size([1, 400])
clean_input_spectrogram.size() == torch.Size([1, 400, 161])
output.size() == torch.Size([1, 400, 161])

mask仅包含0和1。由于这是一个掩码，因此我想将output的相关元素设为clean_input_spectrogram的情况下，将mask的元素设置为等于1。

我该怎么做？

Answer 1

您可以执行以下操作：

• > res [,1] [,2] [,3] [1,] "toys" "345" "1" [2,] "food" "45" "2" 是您的面具；
• m是您的谱图；
• x是您的输出；

o