avg_pool = output.masked_fill(mask[:, :, None], 0).mean(dim=1) ... Sorry i was being silly because my brain couldn't put masked_fill and ...
確定! 回上一頁