你即將離開本站

並前往https://www.thinbug.com/q/47447272

tensorflow的功能是否类似于pytorch＆＃34; masked_fill _＆＃34 ...

我想通过掩码矩阵将INF值设置为矩阵，就像pytorch代码一样： scores.data.masked_fill_(y_mask.data, -float('inf')). 我尝试使用 tf.map_fn 来实现它，但性能太慢了。

確定！ 回上一頁

查詢 「Masked_fill」的人也找了：

  1. Masked_fill
    2.
  2. Torch matmul masked_fill
    3.
  3. Pytorch mask
    4.
  4. PyTorch Tensor 教學
    5.
  5. Python FloatTensor
    6.
  6. Pytorch index mask
    7.
  7. Torch mask fill
    8.
  8. Masked fill
    9.