JLFaller's picture
Fix: explicitly cast to bool tensor before masked scatter
8c1b50a verified