tfa.layers.CRF | TensorFlow Addons
www.tensorflow.org › python › tfaNov 15, 2021 · Consider a Conv2D layer: it can only be called on a single input tensor of rank 4. As such, you can set, in __init__ (): self.input_spec = tf.keras.layers.InputSpec(ndim=4) Now, if you try to call the layer on an input that isn't rank 4 (for instance, an input of shape (2,), it will raise a nicely-formatted error: