Bilinear

Applies a bilinear transformation to the incoming data.

Abstract Signature:

Bilinear(input1, input2, weight, bias)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.bilinear

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '{input1} @ {weight} @ {input2} + {bias}'

NumPy

API: —

Strategy: Macro '{input1} @ {weight} @ {input2} + {bias}'

Keras

API: —

Strategy: Custom / Partial

TensorFlow

API: —

Strategy: Custom / Partial

Apple MLX

API: mlx.nn.layers.linear.Bilinear

Strategy: Macro '{input1} @ {weight} @ {input2} + {bias}'

Flax NNX

API: —

Strategy: Macro '{input1} @ {weight} @ {input2} + {bias}'

PaxML / Praxis

API: —

Strategy: Macro '{input1} @ {weight} @ {input2} + {bias}'