input: Write a python function to perform matrix multiplication using PyTorch.
  assert (matrix_multiplication([[1, 2], [3, 4]], [[5, 6], [7, 8]]) == [[19, 22], [43, 50]]).all()
  assert (matrix_multiplication([[1, 0], [0, 1]], [[1, 2], [3, 4]]) == [[1, 2], [3, 4]]).all()
  assert (matrix_multiplication([[2, 3], [4, 5]], [[1, 2], [3, 4]]) == [[11, 16], [19, 28]]).all()
output: import torch
  
  def matrix_multiplication(A, B):
      tensor_A = torch.tensor(A)
      tensor_B = torch.tensor(B)
      result = torch.matmul(tensor_A, tensor_B)
      return result.numpy().tolist()