PyTorch: indexar con tensores uint8 pasa a ser un error
Útil para
PyTorch deja de tratar los tensores uint8 como máscara booleana al indexar y ahora lanza un IndexError. Ese uso estaba avisado como obsoleto desde la versión 1.2. La migración consiste en convertir la máscara con x[mask.bool()]. El cambio afecta a la indexación, a index e index_put_, y a los kernels meta usados por torch.compile.
Qué significa
El código que indexaba con x[mask] usando una máscara uint8 dejará de funcionar y devolverá un error. Quien lo use tendrá que pasar la máscara a booleana.
Cómo te afecta
- Programadores
Busca en tu código x[mask] con máscaras uint8: ahora lanza IndexError, así que pásalas a x[mask.bool()]. Repasa también index e index_put_ si los usas dentro de torch.compile.
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