backend/lsquant/fast_math/mat_to_vect.py

import torch as tr

def op_ket(op, vect):
    p = tr.einsum("ij,j->i", op, vect)
    return p