terminator.models.layers.gvp.tuple_index

terminator.models.layers.gvp.tuple_index(x, idx)

Indexes into a tuple (s, V) along the first dimension.

Parameters:

idx – any object which can be used to index into a torch.Tensor