「masked_select 函数最关键的参数就是布尔掩码 mask,传入 mask 参数的布尔张量通过 True 和 False (或 1 和 0) 来决定输入张量对应位置的元素是否保留,既然是一一对应的关系,这就需要传入 mask 中的布尔张量和传入 input 中的输入张量形状要相同。」这里需要注意此时的形状相同包括显式的相等,还包括隐式的相等。
[TrueFalseTrueFalse] 将输入张量和广播后的布尔张量一一对应,通过 True 和 False 决定是否筛选出该元素,最终筛选出来的元素为 0 和 2,由于使用 masked_select 函数返回的都是 1D 张量,因此最终的结果为 tensor([0, 2])。 importtorch# mask.size() ≠ x.size()但是mask能够广播成x.size()的形状x=to...