original_docstring: 
      Convert rotation matrices to angle-axis format.
      Args:
          oris: np array of shape (seq_length, n_joints*9).
      Returns: np array of shape (seq_length, n_joints*3)
      
docstring: Convert rotation matrices to angle-axis format.
docstring_tokens: Convert rotation matrices to angle - axis format .
code: def rotmat2aa(rotmats):
      """
      Convert rotation matrices to angle-axis format.
      Args:
          oris: np array of shape (seq_length, n_joints*9).
      Returns: np array of shape (seq_length, n_joints*3)
      """
      seq_length = rotmats.shape[0]
      assert rotmats.shape[1] % 9 == 0
      n_joints = rotmats.shape[1] // 9
      ori = np.reshape(rotmats, [seq_length*n_joints, 3, 3])
      aas = np.zeros([seq_length*n_joints, 3])
      for i in range(ori.shape[0]):
          aas[i] = np.squeeze(cv2.Rodrigues(ori[i])[0])
      return np.reshape(aas, [seq_length, n_joints*3])
code_tokens: def rotmat2aa ( rotmats ) : seq_length = rotmats . shape [ 0 ] assert rotmats . shape [ 1 ] % 9 == 0 n_joints = rotmats . shape [ 1 ] // 9 ori = np . reshape ( rotmats , [ seq_length * n_joints , 3 , 3 ] ) aas = np . zeros ( [ seq_length * n_joints , 3 ] ) for i in range ( ori . shape [ 0 ] ) : aas [ i ] = np . squeeze ( cv2 . Rodrigues ( ori [ i ] ) [ 0 ] ) return np . reshape ( aas , [ seq_length , n_joints * 3 ] )
short_docstring: Convert rotation matrices to angle-axis format.
short_docstring_tokens: Convert rotation matrices to angle - axis format .
comment: """
      Convert rotation matrices to angle-axis format.
      Args:
          oris: np array of shape (seq_length, n_joints*9).
      Returns: np array of shape (seq_length, n_joints*3)
      """