説明
TensorFlowでn×mの行列の値をすべて1にするにはones()を使います。ones()の最初のパラメーターに生成する行列の数を指定します。2番目の以降のパラメーターは省略することができます。サンプルプログラム [サンプルをダウンロード]
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.ones([4,7]); ss= tf.Session(config=config) print(ss.run(a))
実行結果
