在 PyTorch 中使用 index_select()

DEV Community · 2024-09-25T14:22:28Z

文章介绍了如何使用 PyTorch 的 `index_select()` 函数从张量中选择元素。该函数需要输入张量、维度和索引三个参数，适用于 0D 或 1D 张量，并返回选定元素的张量。示例展示了在不同维度上选择元素的灵活性。