terminator.models.layers.gvp.tuple_cat

terminator.models.layers.gvp.tuple_cat(*args, dim=- 1)[source]

Concatenates any number of tuples (s, V) elementwise.

Parameters:

dim – dimension along which to concatenate when viewed as the dim index for the scalar-channel tensors. This means that dim=-1 will be applied as dim=-2 for the vector-channel tensors.