input: def optimize_transpose(A):
      """ Optimizes a Python function that processes a 2d array and stores the transpose output in a new array.
      Example:
      >>> optimize_transpose([[1, 2, 3], [4, 5, 6]])
      [[1, 4], [2, 5], [3, 6]]
      """
output:     rows = len(A)
      cols = len(A[0])
      transposed = [[0] * rows for _ in range(cols)]
      for i in range(rows):
          for j in range(cols):
              transposed[j][i] = A[i][j]
      return transposed