人工智能资料TensorFlow A system for large_scale machine learning.pdf

TensorFlow: A system for large-scale machine learning ´ Martın Abadi, Paul Barham, Jianmin Chen, Zhifeng Chen, Andy Davis, Jeffrey Dean, Matthieu Devin, Sanjay Ghemawat, Geoffrey Irving, Michael Isard, Manjunath Kudlur, Josh Levenberg, Rajat Monga, Sherry Moore, Derek G. Murray, Benoit Steiner, Paul Tucker, Vijay Vasudevan, Pete Warden, Martin Wicke, Yuan Yu, and Xiaoqiang Zheng 6 Google Brain 1 0 2 y Abstract We introduce the TensorFlow system1 for experiment- a ing with new models, training them on large datasets, and M TensorFlow is a machine learning system that operates at moving them into production. We have based TensorFlow large scale and in heterogeneous environments. Tensor- on years of experience with our first-generation system, 7 2 Flow uses dataflow graphs to represent computation, DistBelief [21], both simplifying and generalizing it to en- shared state, and the operations that mutate that state. It able researchers to explore a wider variety of ideas with ] maps the nodes of a dataflow graph across many machines relative ease. TensorFlow supports both large-scale train- C in a cluster, and within a machine across multiple com- ing and inference: it efficiently uses hundreds of powerful D putational devices, including multicore CPUs, general- (GPU-enabled) servers for fast training, and it runs trained s. purpose GPUs, and custom designed ASICs known as models for inference in production on various platforms, c Tensor Processing Uni