如何创建一个函数来计算时间差(以numpy数组的形式)

import numpy as np

def calculate_time_difference(start_time, end_time):
    time_diff = end_time - start_time
    return time_diff

start_time = np.array(['2022-01-01 12:00:00', '2022-01-02 09:30:00'], dtype='datetime64')
end_time = np.array(['2022-01-01 13:30:00', '2022-01-02 10:45:00'], dtype='datetime64')

time_difference = calculate_time_difference(start_time, end_time)
print(time_difference)