kerasでbackend その5

概要

kerasでbackendで、fizzbuzz書いてみた。

サンプルコード

from tensorflow.contrib.keras.python.keras import backend as K
import tensorflow as tf

fizz = K.constant("Fizz", dtype = tf.string)
buzz = K.constant("Buzz", dtype = tf.string)
fizzbuzz = K.constant("FizzBuzz", dtype = tf.string)

def fn(x):
    return K.switch(tf.equal(tf.mod(x, 15), 0), fizzbuzz, K.switch(tf.equal(tf.mod(x, 5), 0), buzz, K.switch(tf.equal(tf.mod(x, 3), 0), fizz, tf.as_string(x))))

kvar = K.arange(1, 100, step = 1, dtype = 'int32')
map = K.map_fn(fn, kvar, name = None, dtype = tf.string)
print (K.eval(map))

以上。