如何在C++中，获取三维torch::Tensor的各维度的大小

在C++中，可以通过以下代码获取三维torch::Tensor的各维度的大小：

torch::Tensor tensor = torch::rand({3, 4, 5});  // 创建一个大小为 3x4x5 的 Tensor
int64_t dim_0 = tensor.size(0);  // 获取第一维度的大小，即 3
int64_t dim_1 = tensor.size(1);  // 获取第二维度的大小，即 4
int64_t dim_2 = tensor.size(2);  // 获取第三维度的大小，即 5

其中，`size()`函数返回的是一个`std::vector<int64_t>`类型的向量，表示各维度的大小。通过使用`[]`运算符可以获取指定维度的大小。