tf.broadcast_toを各バッチごとに行いたい - Qiita

はじめに（やりたいこと） 最近はpytorchから離れ、tensorflowとお友達になるべく日々奮闘しています。 もちろんpytorchのことを嫌いになったわけではないですよ。 行列演算をしているときにtf.broadcast_toでtensorをいい感じにしたいことがあります。でも、もしそのtensorがバッチだった場合…… >>> x = tf.constant([[1, 2, 3, 4, 5], [6, 7, 8, 9, 0]]) >>> tf.broadcast_to(x, [2, 5, 5]) Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/home/***/python3.7/site-packages/tensorflow/python/ops/gen_array_o