旋转张量的操作是什么？

Question

在我的情况下，对于2x2矩阵（张量），我需要以下操作：

a = [[1,2],
     [3,4]]

输出为b=np.rot90(np.rot90(a))

b = [[4,3],
     [2,1]]

有人知道这个操作吗？ b = sess.run(tf.reverse(a,axis=-1))似乎不是那个。感谢。

更新

我知道@jakeoung建议的操作tf.contrib.image.rotate，但是这个操作是否支持反向prorogation？

Answer 1

您可以使用tf.image.rotation功能。参见：

import tensorflow as tf

a = tf.constant([[1,2],
     [3,4]], dtype=tf.float32)
# a_ = tf.expand_dims(a, 2)
# print(a_)
b = tf.contrib.image.rotate(tf.contrib.image.rotate(a, 90), 90)

with tf.Session() as sess:
  [rot] = sess.run([b])
  print(rot)