ei_dot_ej.py

import numpy
import perfplot


def setup(n):
    pts = numpy.random.rand(3, n, 2)
    e = numpy.array([
        pts[2] - pts[1],
        pts[0] - pts[2],
        pts[1] - pts[0],
    ])
    return e


def test1(data):
    numpy.einsum("...k, ...k->...", data, data)
    w = numpy.einsum("...k, ...k->...", data[[1, 2, 0]], data[[2, 0, 1]])
    return w


def test2(data):
    v = numpy.einsum("...k, ...k->...", data, data)
    w = v - numpy.sum(v, axis=0) / 2
    return w


perfplot.save(
    "out.png",
    setup=setup,
    kernels=[test1, test2],
    n_range=[2 ** k for k in range(23)],
)