TensorFlowを使用して1つのレイヤーで並列線形計算を作成する方法は？

bugfix> tensorflow > 投稿 2020-08-22 18:59

サイズが[batch_size、height、width]の入力があります。ここでは、1つのレイヤーでいくつかの異なる並列線形変換を行います。

x = tensor([batch_size, height, width])
y = [W1*x, W2*x, W3*x,...,Wn*x]

fully_connected があることに気づいたおよび layer.dense TensorFlowではありますが、一度に1つの線形変換しか行えませんか？それらを使用して並列線形変換を実行できますか？

私はTensorFlowを初めて使用しますが、この質問がちょっとばかげているとすみません。

tensorflow

回答 1 件

ブロードキャストを活用する：

import tensorflow as tf
batch_size, height, width = 5, 4, 3
n = 2
x = tf.random_uniform((batch_size, height, width))
W = tf.random_uniform((n,))
y = tf.multiply(tf.reshape(W, (n, 1, 1, 1)), tf.expand_dims(x, 0))
with tf.Session() as sess:
    y = sess.run(y)
    print(y.shape)
    # (2, 5, 4, 3)

Trends