def sort_rows()

in swd.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def sort_rows(matrix, num_rows):
    matrix_T = array_ops.transpose(matrix, [1, 0])
    sorted_matrix_T = nn_ops.top_k(matrix_T, num_rows)[0]
    return array_ops.transpose(sorted_matrix_T, [1, 0])