File size: 80 Bytes
1c703f0
 
 
 
 
1
2
3
4
5

import  torch as tr
def bra_ket(a,b):
    c=tr.einsum("i,i->",a,b)
    return c