int类型的张量来索引张量
bool类型的张量来索引张量
我们创建了一个包含了从 0 到 9 的整数的张量 x。然后,我们创建了一个 bool 类型的张量 bool_indices,其中对应位置为 True 的元素表示选择该位置的元素,而 False 表示不选择。最后,我们使用 bool_indices 张量作为索引来选择 x 中对应位置为 True 的元素,并将结果保存在 selected_elements 中
用张量来索引张量
于 2024-02-27 22:51:02 首次发布