Created by: chenwthu

如何对Tensor实现类似numpy中使用range作为索引的操作，例如np.array([[1,2,3], [4,5,6]])[range(2), [1,2]]可以得到[2,6]