TensorFlow je komplexní open-source platforma pro strojové učení (Machine Learning), kterou vyvinul tým Google Brain. Je navržena tak, aby usnadnila proces od výzkumu až po nasazení modelů do reálného provozu na různých zařízeních – od výkonných serverových clusterů až po chytré telefony a IoT zařízení.
Název je odvozen od operací, které neuronové sítě provádějí nad vícedimenzionálními datovými poli, zvanými tenzory.
TensorFlow se postupem času vyvinul v ucelený ekosystém, který zahrnuje:
Keras je oficiální rozhraní pro TensorFlow, které umožňuje definovat a trénovat neuronové sítě pomocí pár řádků kódu. Je velmi intuitivní a zaměřené na uživatelskou přívětivost.
TensorFlow vyniká v tom, co se stane s modelem *po* trénování:
Unikátní nástroj pro vizualizaci trénovacího procesu. Umožňuje sledovat metriky (přesnost, ztrátu), prohlížet strukturu výpočetního grafu a analyzovat histogramy vah v reálném čase.
TensorFlow prošel zásadní transformací mezi verzemi 1.x a 2.x:
| Vlastnost | TensorFlow | PyTorch |
|---|---|---|
| Hlavní vývojář | Meta (Facebook) | |
| Produkční nasazení | Vynikající (TFX, TF Serving) | Velmi dobré (TorchScript) |
| Mobilní/Edge zařízení | Lídr trhu (TF Lite) | Rychle se dotahuje (ExecuTorch) |
| Styl kódu | Integrovaný Keras (velmi strukturovaný) | Čistý Python (velmi flexibilní) |
| Vizualizace | TensorBoard (v základu) | Externí nástroje (Visdom, WandB) |
TensorFlow je optimalizován pro širokou škálu hardwaru:
TensorFlow je ideální volbou, pokud plánujete model nasadit do rozsáhlé produkce, potřebujete jej spustit na mobilním telefonu, nebo pokud využíváte cloudovou infrastrukturu Google Cloud, která je pro TensorFlow nativně optimalizována.
— Související termíny: Keras, PyTorch, TPU, Strojové učení, ONNX.