Method coords_of_step

physo/physym/vect_programs.py:710–726

Helper method returning the tuple of coordinates corresponding to a step. Parameters ---------- step : int Step. Returns ------- coords : numpy.array of shape (2, batch_size,) of int Coordinates, 0th array in batch dim

(self, step)

        return np.tile(batch_vect, (self.max_time_step, 1)).transpose()  # (batch_size, max_time_step,) of ?

    def coords_of_step (self, step):
        """
        Helper method returning the tuple of coordinates corresponding to a step.
        Parameters
        ----------
        step : int
            Step.
        Returns
        -------
        coords : numpy.array of shape (2, batch_size,) of int
            Coordinates, 0th array in batch dim and 1th array in time dim.
        """
        coords = np.stack((                   # (2, batch_size,) of int
            self.tokens.pos_batch [:, step],  # batch dim coord
            self.tokens.pos       [:, step],  # time dim coord
            ), axis=0)
        return coords

    # -----------------------------------------------------------------------------------------------------------------
    # -------------------------------------------- UTILS : TOKEN MANAGEMENT -------------------------------------------