説明
TensorFlowで四捨五入するにはround()を使います。round()の最初のパラメーターには数値や行列を指定します。サンプルプログラム [サンプルをダウンロード]
import tensorflow as tf import multiprocessing as mp core_num = mp.cpu_count() config = tf.ConfigProto( inter_op_parallelism_threads=core_num, intra_op_parallelism_threads=core_num ) a = tf.round([0.12, 0.5, 0.6, 3.14, -2.2, -2.5, -2.6]) ss= tf.Session(config=config) print(ss.run(a))
実行結果
