python - テンソルフローでのバッチ画像拡張

Question

たとえば、テンソルフローでランダムな角度を回転させます。各バッチで、画像ごとに異なるランダムな角度を回転させたいです。ランダムに生成された角度テンソルを使用tf.contrib.image.rotateしてそれを行うことができます:image_batch

radian = tf.random_uniform(
    (batch_size),  
    minval=-ROT_TH,
    maxval=ROT_TH,
    dtype=tf.float32,
    seed=None,
    name=None
)
rotated_batch = tf.contrib.image.rotate(image_batch, radian)

ただし、 でバッチをビルドするとallow_smaller_final_batch=True、はバッチ サイズが固定されないbatch_sizeため、 は役に立ちません。image_batchまた、radian と image_batch の N 次元が同じではないため、回転は失敗します。

どうすれば修正できますか？