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)
def _multi_deref ( tensors : List [ tf . Tensor ] , index : tf . Tensor ) -> List [ tf . Tensor ] : 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 )
Equivalent to `[t[index, ...] for t in tensors]`.

    See `_deref` for more details.