如何高效选取TorchLua数组或张量的任意子集？

本文共计379个文字，预计阅读时间需要2分钟。

我正在使用Python/Lua，并拥有一个包含10个元素的数组数据集。数据集如下：dataset={11, 12, 13, 14, 15, 16, 17, 18, 19, 20}。如果我想读取数组中的第1个元素，我可以直接访问dataset[1]。我只需要选择数组中的前3个元素。