c++ libtorch 创建多维单位张量

原创已于 2024-10-22 17:04:11 修改 · 258 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

于 2024-10-12 20:41:59 首次发布

torch 专栏收录该内容

21 篇文章

订阅专栏

3维单位张量，代码示例

torch::Tensor multi_dim_identity = torch::zeros({ 2, 2, 2 }, torch::kComplexDouble);
for (int i = 0; i < 2; ++i) {
    multi_dim_identity.index_put_({ torch::indexing::Slice(), i, i }, 1);
}

torch::indexing::Slice()代表第一维度的所有元素

4维单位张量，代码示例

torch::Tensor multi_dim_identity = torch::zeros({ 2, 2， 2, 2 }, torch::kComplexDouble);
for (int i = 0; i < 2; ++i) {
    multi_dim_identity.index_put_({ torch::indexing::Slice(), torch::indexing::Slice(), i, i }, 1);
}

第一个torch::indexing::Slice()代表第一维度的所有元素，第2个torch::indexing::Slice()代表第2维度的所有元素