... value=10)) # masked_fill, 和masked_select类似,赋值value到被选的index中,得到新的张量print(torch.masked_fill(r, r.lt(10), value=100))
確定! 回上一頁