如何在libtorch中将形状(n，k)的张量与形状(k)的张量叠加？

#include <torch/torch.h>

int main() {
    // 创建形状为(n, k)的张量
    int n = 3;
    int k = 4;
    torch::Tensor tensor1 = torch::ones({n, k});

    // 创建形状为(k)的张量
    torch::Tensor tensor2 = torch::ones({k});

    // 将形状为(k)的张量扩展为形状为(1, k)的张量
    tensor2 = tensor2.unsqueeze(0);

    // 将形状为(1, k)的张量扩展为形状为(n, k)的张量
    tensor2 = tensor2.expand_as(tensor1);

    // 将两个张量进行叠加
    torch::Tensor result = tensor1 + tensor2;

    // 打印结果
    std::cout << result << std::endl;

    return 0;
}