docstr: Check that the output of convolution has the correct shape.
code: def test_feature_steered_convolution_output_shape(
      self,
      batch_size,
      num_vertices,
      in_channels,
      out_channels,
      num_weight_matrices,
  ):
      """Check that the output of convolution has the correct shape."""
      data, neighbors = _dummy_data(
          batch_size, num_vertices, in_channels
      )
      u, v, c, w, b = _dummy_variables(
          in_channels, out_channels, num_weight_matrices
      )
  
      y = gc.feature_steered_convolution(
          data=data,
          neighbors=neighbors,
          sizes=None,
          var_u=u,
          var_v=v,
          var_c=c,
          var_w=w,
          var_b=b,
      )
      y_shape = y.shape.as_list()
  
      self.assertEqual(y_shape[-1], out_channels)
      self.assertAllEqual(y_shape[:-1], data.shape[:-1])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import itertools
  from absl.testing import parameterized
  import numpy
  from six.moves import range
  import tensorflow
  import tensorflow_graphics.geometry.convolution.graph_convolution
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 4 for calling the GraphConvolutionTestFeatureSteeredConvolutionTests obj's underlying member method code with example usage: obj.test_feature_steered_convolution_output_shape(batch_size, num_vertices, in_channels, out_channels, num_weight_matrices) without return types