canonical_trafo.py 文件源码

def _get_rot_mat_z(angle):
    """ Returns a 3D rotation matrix. """
    one_vec = tf.ones_like(angle)
    zero_vec = one_vec*0.0
    trafo_matrix = _stitch_mat_from_vecs([tf.cos(angle), tf.sin(angle), zero_vec,
                                          -tf.sin(angle), tf.cos(angle), zero_vec,
                                          zero_vec, zero_vec, one_vec])
    return trafo_matrix