TensorFlow：float32テンソルをfloat32にキャスト

Question

私は使用しようとしています：train = optimizer.minimize(loss)標準オプティマイザはtf.float64で動作しません。したがって、lossをtf.float64からtf.float32に切り捨てたいと思います。

Traceback (most recent call last): 
    File "q4.py", line 85, in <module> 
    train = optimizer.minimize(loss) 
    File "/Library/Python/2.7/site-packages/tensorflow/python/training/optimizer.py", line 190, in minimize 
    colocate_gradients_with_ops=colocate_gradients_with_ops) 
    File "/Library/Python/2.7/site-packages/tensorflow/python/training/optimizer.py", line 229, in compute_gradients 
    self._assert_valid_dtypes([loss]) 
    File "/Library/Python/2.7/site-packages/tensorflow/python/training/optimizer.py", line 354, in _assert_valid_dtypes 
    dtype, t.name, [v for v in valid_dtypes])) 
ValueError: Invalid type tf.float64 for Add_1:0, expected: [tf.float32]. 

Answer 1

短い答えは、あなたがtf.cast()オペアンプを使用してtf.float64からtf.float32にテンソルを変換することができるということです。

loss = tf.cast(loss, tf.float32) 

長い答えは、これはオプティマイザを使用してすべての問題を解決することはできませんということです。 （tf.float64のサポートの欠如は、known issueです。）オプティマイザは、最適化しようとしているtf.Variableオブジェクトのタイプもすべてtf.float32である必要があります。