在pytorch中使用cuda完成两个向量相加

可以使用以下代码来完成两个向量的相加：

import torch

# 定义两个向量
vec1 = torch.Tensor([1, 2, 3])
vec2 = torch.Tensor([4, 5, 6])

# 将向量转换为cuda tensor
vec1_cuda = vec1.cuda()
vec2_cuda = vec2.cuda()

# 将两个cuda tensor相加
result_cuda = vec1_cuda + vec2_cuda

# 将结果转换为cpu tensor并打印
result = result_cuda.cpu()
print(result)

这段代码首先定义了两个向量`vec1`和`vec2`，然后将它们转换为CUDA tensor `vec1_cuda`和`vec2_cuda`。接着，使用加号运算符将两个CUDA tensor相加，得到结果CUDA tensor `result_cuda`。最后，将结果CUDA tensor转换为CPU tensor并打印结果。