下記のコードのように、tensorflowで、random_normal で初期化した行列に足し算をしたらうまくブロードキャストされないのですが、理由がよくわかりません。
Aが 3 x 3の行列で、 Bが 3 x 1の行列です。
どなたか教えていただけますと幸いです。
import tensorflow as tf import numpy as np A = tf.random_normal([3, 3], mean=1, stddev=4, seed = 1) B = tf.Variable([[1, 2, 3]], dtype=tf.float32) C = A + B init = tf.initialize_all_variables() with tf.Session() as sess: sess.run(init) print("A") print(A.eval()) print("\nB") print(B.eval()) print("\nC") print(C.eval())
以下実行結果
A [[-2.24527287 6.93839502 1.26131749] [-8.77081585 1.39699364 3.36489725] [ 3.37129188 -7.49171829 -1.89158893]] B [[ 1. 2. 3.]] C [[ 6.96361637 4.79503536 5.32562923] [ 2.75787783 2.64941168 2.83947849] [ 1.70853901 9.26744461 1.85234141]]
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2018/06/02 23:52