python - 如何在 Tensorflow 中创建旋转矩阵

Question

我想在 tensorflow 中创建一个旋转矩阵，其中所有部分都是张量。

我有什么:

def rotate(tf, points, theta):
    rotation_matrix = [[tf.cos(theta), -tf.sin(theta)],
                       [tf.sin(theta), tf.cos(theta)]]
    return tf.matmul(points, rotation_matrix)

但这表示 rotation_matrix 是张量列表而不是张量本身。 theta 也是运行时传入的张量对象。

Answer 1

有两个操作:

def rotate(tf, points, theta):
    rotation_matrix = tf.pack([tf.cos(theta),
                              -tf.sin(theta),  
                               tf.sin(theta),
                               tf.cos(theta)])
    rotation_matrix = tf.reshape(rotation_matrix, (2,2))
    return tf.matmul(points, rotation_matrix)