fn
bilinear_layer_op
→TensorImplPtrint bilinear_layer_op(const int & x1, const int & x2, const int & weight, const int & bias)Public bilinear-layer entry point.
Thin wrapper that delegates to BilinearLayerBackward::forward.
Parameters
x1TensorImplPtrFirst operand of the bilinear form (last dim ).
x2TensorImplPtrSecond operand of the bilinear form (last dim ).
weightTensorImplPtrWeight tensor of shape
(D_out, D_1, D_2).biasTensorImplPtrOptional bias vector of shape
(D_out,); pass nullptr to omit.Returns
TensorImplPtrOutput tensor of shape (*, D_out).