def METHOD_NAME():
    def get_tensor(si):
        return np.arange(si.idx_in_epoch, si.idx_in_epoch + shape[0], dtype=np.int32)
    inp = fn.external_source(get_tensor, batch=False)
    return inp.gpu()