dot vs einsum for 1D

dot-vs-einsum.py

import numpy as np
import perfplot


def dot(xy):
    x, y = xy
    return np.dot(x, y)


def einsum(xy):
    x, y = xy
    return np.einsum("i...,i...->...", x, y)


b = perfplot.bench(
    setup=lambda n: np.random.rand(2, n),
    kernels=[dot, einsum],
    n_range=[2 ** k for k in range(25)],
)
b.save("out.svg")
b.show()