SubtractLayer

Layer that performs elementwise subtraction on a list of two inputs.

PyTorchJAX (Core)KerasTensorFlow

PyTorch

API: —

Strategy: Plugin (layer_subtract_shim)

JAX (Core)

API: —

Strategy: Plugin (layer_subtract_shim)

Keras

API: keras.layers.Subtract

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.keras.layers.Subtract

Strategy: Direct Mapping

Official Docs ↗