PyTorch: masked_scatter_ solo acepta máscaras booleanas en MPS y meta
Útil para
Cambio en PyTorch que hace que masked_scatter_ en los backends MPS y meta rechace máscaras uint8, igual que ya ocurre en CPU y CUDA. Antes, MPS y meta aceptaban estas máscaras mediante una ruta obsoleta. El cambio es incompatible y la solución es convertir la máscara con mask.bool(). También se comprueba el tipo de la máscara antes de devolver resultados para entradas vacías.
Qué significa
El código que pasaba máscaras uint8 a masked_scatter_ en Mac (MPS) o en el modo meta dejará de funcionar hasta que las máscaras se conviertan a booleanas con mask.bool().
Fuentes originales
Resumen de Hoy en IA. Pulsa una fuente para leer la noticia completa en su idioma original.
- GitHub pytorch/pytorchLeer la noticia completa en GitHub pytorch/pytorch