本篇啟用 tensorflow  GPU 計算 pi 值，利用 tf.sqrt 及 tf.square 計算大量張量的平方及開根號求取與原點的距離。此張量的大小每次 (batch) 為 1 億個點，然後計算 50 萬次。也就是說計算 50 萬 * 1 億個點。GPU產生 1 億個亂數點非常快速，計算 1 億個點與原點的距離也只需 0.1 秒，比CPU快了很多。

 GPU 一啟動，顯卡的溫度立即上升到攝氏 77 度，非常耗電。

完整代碼

底下為蒙地卡羅 tf 版的完整代碼。