pytorch中的index_select函数

index_select(dim, index):

dim: 按行/列进行索引，dim=0表示按行，dim=1表示按列。
index: 是一个tensor类型，用于索引行/列数值的一个列表，例如[1，2]，dim=0,表示按行索引1和2行值。[1,3,5]，dim=0，表示按行索引第1，3，5行。
备注： 可以使用index_select()函数生成batch_size大小的样本（例如在线性回归中，自定义一个读取batch_size大小的函数，每次返回batch_size大小的样本和标签）。