PyTorch — Python-библиотека для тензорных вычислений с GPU-ускорением, аналогично NumPy. Она предоставляет ленточную систему автоматического дифференцирования (autograd) для построения глубоких нейронных сетей.
Ключевая особенность PyTorch — динамические нейронные сети на основе tape-based autograd. Это позволяет менять архитектуру на ходу без перестроения графа. Библиотека глубоко интегрирована с Python, упрощая отладку и написание кода.
PyTorch состоит из компонентов: torch, torch.autograd, torch.nn, torch.jit, torch.multiprocessing и torch.utils. Это замена NumPy на GPU и гибкая платформа для исследований.
GitHub ★ 101,049
0 комментариев