bugfix> tensorflow > 投稿

サイズが[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を初めて使用しますが、この質問がちょっとばかげているとすみません。

回答 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)
    
    

あなたの答え