TensorFlowの写像

Question

x=tf.Variable(tf.ones(3))をW=tf.Variable(tf.ones((1, 5)))で写像して(3, 5)のshapeを持ったテンソルxWにしたいのですが、うまくいきません。
xのshapeが(3, 1)ではなく(3, )となっていることが原因だと思いますが、何か良い方法は無いでしょうか？

Comments

[mokemokechicken]

そのx, Wを使って、どういう計算をしたいのでしょうか。直積か何かですか？
https://ja.wikipedia.org/wiki/%E7%9B%B4%E7%A9%8D_(%E3%83%99%E3%82%AF%E3%83%88%E3%83%AB)

[opeco17]

(3, 1)×(1, 5) = (3, 5)となるような行列の積を計算したいです。

Answer 1

numpy 同様、演算 * は要素ごとの積になるので、行列積 (matrix multiplication) を計算したい場合は tf.matmul または @ 演算子をお使いください。

形状が (3,) と (1, 5) のテンソル同士では行列積が計算できないので、tf.reshape で (3, 1) に変更した上で計算を行います。

import numpy as np
import tensorflow as tf


x = tf.Variable(tf.ones(3))
W = tf.Variable(tf.ones((1, 5)))
y = tf.matmul(tf.reshape(x, (3, 1)), W)
# y = tf.reshape(x, (3, 1)) @ W でもよい。

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())

    ret = sess.run(y)

print(ret)
# [[1. 1. 1. 1. 1.]
#  [1. 1. 1. 1. 1.]
#  [1. 1. 1. 1. 1.]]