「FizzBuzzを書いてください」「まずTensorFlowをインポートして・・・」を自分も書いてみた

下のツイートとかで有名になったFizzBuzz-TensorFlowネタ。今日もちょっと見かけたので思いつきで書いてみた。

IT会社面接官：「数字を列挙し、3の倍数ならfizz、5の倍数ならbuzz、15の倍数ならfizzbuzzを出力するプログラムを書いてください。」
面接を受けている人：「では、まずTensorFlowをインポートします…」https://t.co/UNk2jH5rfn

— Graham Neubig (@neubig) 2016年5月24日

実装はたくさんあって「FizzBuzz TensorFlow」でウェブ検索すればたくさん出てきます。最適化器で機械学習として解くのが多いけれどTensorFlowは「数値微分による最適化可能な実行可能な計算グラフを作成するフレームワーク」であり、かつ論理演算のオペレーションも文字列操作のオペレーションも備えているため素直に実装できます。

「そこらへんのディープラーニングしかできないフレームワークとは違うのだよ！！！」

API調べながら思い出しながら10分位で書けました。素直なバージョンもウェブ検索で結構見つかりますがtf.caseとかtf.control_dependenciesあたりでイロイロ凝ったことできるというのは覚えておいてください。

import tensorflow as tf

a = tf.Variable(0, name='counter')

with tf.control_dependencies([tf.assign_add(a, 1)]):
    b = tf.case({
        tf.equal(tf.mod(a, 15), 0): lambda: tf.constant("FizzBuzz"),
        tf.equal(tf.mod(a, 3), 0): lambda: tf.constant("Fizz"),
        tf.equal(tf.mod(a, 5), 0): lambda: tf.constant("Buzz"),
        }, default=lambda: tf.as_string(a))

sess = tf.Session()
sess.run(tf.global_variables_initializer())

for i in range(100):
    print(sess.run(b).decode())