chumpy/reordering.py

import numpy as np

from .ch import Ch


class Select(Ch):
    @property
    def r(self):
        base = np.asarray(self.a).reshape(-1)
        selected = base[np.asarray(self.idxs, dtype=np.int64)]
        return selected.reshape(self.preferred_shape)