tfg.geometry.convolution.utils.check_valid_graph_pooling_input

View source on GitHub

Checks that the inputs are valid for graph pooling.

Note:

In the following, A1 to An are optional batch dimensions.

data A float tensor with shape [A1, ..., An, V1, C].
pool_map A SparseTensor with the same type as data and with shape [A1, ..., An, V2, V1].
sizes An int tensor of shape [A1, ..., An, 2]. Can be None.

TypeError if the input types are invalid.
ValueError if the input dimensions are invalid.