pytorch masked_fill方法理解. 9 个月前· 来自专栏python桃花源. mask必须是一个ByteTensor 而且shape的最大维度必须和a一样并且元素只能是0或者1 ,
確定! 回上一頁