tfg.geometry.convolution.utils.check_valid_graph_unpooling_input

View source on GitHub

Checks that the inputs are valid for graph unpooling.

Note:

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

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

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