クロの制作日記

2020-04-28から1日間の記事一覧

Tensorflowのwhile_loopで配列に値を結合する方法

最初に tf.while_loopで以下のように配列に値を結合していくとエラーが吐かれます。 def add(x): return x+1 def condition(_i, x): return _i < 4 def update(_i, x): x = tf.concat([x, [_i]],axis=0) return _i+1, x ind = tf.constant(0) v1 = tf.Variab…

tf.condで引数ありの関数を使いたい場合

Tensorflow1.x系でif文のような条件分岐を実装したい場合はtf.condを利用します(Tensorflow2.x系でも使用可能)。 www.tensorflow.orgtf.condは以下のような引数を渡してあげる必要があります。 tf.cond( pred # 条件式(直接True or Falseを指定可能) true_fn…