]> git.sesse.net Git - voxel-flow/blobdiff - voxel_flow_model.py
Update for TensorFlow 1.0 API.
[voxel-flow] / voxel_flow_model.py
index d419acf6a4493cd534c881c60fbee73bd2e57bff..b3f554d59f935ef327a8a1b86d936461988cb779 100755 (executable)
@@ -85,6 +85,6 @@ class Voxel_flow_model(object):
 
     mask = 0.5 * (1.0 + mask)
     mask = tf.tile(mask, [1, 1, 1, 3])
-    net = tf.mul(mask, output_1) + tf.mul(1.0 - mask, output_2)
+    net = tf.multiply(mask, output_1) + tf.multiply(1.0 - mask, output_2)
 
     return net