Pytorch Vector Dot Product