Pytorch Vector Dot Product Calculator