tfg.geometry.convolution.utils.check_valid_graph_convolution_input

View source on GitHub

Checks that the inputs are valid for graph convolution ops.

Note:

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

data A float tensor with shape [A1, ..., An, V1, V2].
neighbors A SparseTensor with the same type as data and with shape [A1, ..., An, V1, V1].
sizes An int tensor of shape [A1, ..., An]. Optional, can be None.

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