Extract values from a 1-D numpy array for a batch of indices. :param arr: the 1-D numpy array or a func. :param timesteps: a tensor of indices into the array to extract. :param broadcast_shape: a larger shape of K dimensions with the batch dimension equa
(arr_or_func, timesteps, broadcast_shape)
source not stored for this graph (policy: none)