input: Write a python function to perform matrix multiplication.
  assert matrix_multiplication([[1, 2], [3, 4]], [[2, 0], [1, 2]]) == [[4, 4], [10, 8]]
  assert matrix_multiplication([[1, 2, 3], [4, 5, 6]], [[7, 8], [9, 10], [11, 12]]) == [[58, 64], [139, 154]]
  assert matrix_multiplication([[2, 3], [0, 1], [4, 5]], [[1, 2, 3], [4, 5, 6]]) == [[14, 19, 24], [4, 5, 6], [24, 34, 44]]
output: def matrix_multiplication(A, B):
      result = [[sum(a * b for a, b in zip(A_row, B_col)) for B_col in zip(*B)] for A_row in A]
      return result