code: def compute_output_shape(self, input_shape):
      return (*input_shape[:2], self.out_dim)
  
file_path: lambda-networks/lambda_networks/tfkeras.py
imports: import tensorflow
  from einops.layers.tensorflow import Rearrange
  from tensorflow.keras.layers import Conv2D, BatchNormalization, Conv3D, ZeroPadding3D, Softmax, Lambda, Add, Layer
  from tensorflow.keras import initializers
  from tensorflow import einsum, nn, meshgrid
label_desc: Use image node_id 3 for calling the LambdaLayer obj's underlying member method code with example usage: obj.compute_output_shape(input_shape) and returns: unknown,  self