Equivalent to `[t[index, ...] for t in tensors]`.

    See `_deref` for more details.

def _multi_deref(tensors: List[tf.Tensor], index: tf.Tensor) -> List[tf.Tensor]:
    """Equivalent to `[t[index, ...] for t in tensors]`.

    See `_deref` for more details.
    """
    assert tensors
    assert tensors[0].shape[0] > 0
    return _deref_helper(lambda i: [tensor[i, ...] for tensor in tensors],
                         index,
                         0,
                         tensors[0].shape[0] - 1)