File size: 94 Bytes
1c703f0
 
 
 
 
1
2
3
4
5
6
import torch as tr

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