def METHOD_NAME(self, height, width, batch_size):
    grid = self._make_grid(height, width)
    grid = K.flatten(grid)
    grids = K.tile(grid, K.stack([batch_size]))
    return K.reshape(grids, (batch_size, 3, height * width))