For your specified sizes, I get these numbers on a CPU:
# your method
402 µs ± 26.5 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

# my suggestion
115 µs ± 7.22 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)