Как создать матрицу вращения в тензорном потоке

Question

Я хочу создать матрицу вращения в тензорном потоке, где все ее части являются тензорами.

Что у меня есть:

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) 

Answer 2

является вариант, я обнаружил, что работает, чтобы использовать пакет, но если есть лучший способ, пожалуйста, отправить ответ:

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

Answer 3

Чтобы ответить на вопрос " Как построить матрицу вращения », следующее более чистое, чем требуемое несколько вызовов pack (stack):

tf.stack([(tf.cos(angle), -tf.sin(angle)), (tf.sin(angle), tf.cos(angle))], axis=0)