#PyTorch #tensor operation #dot product