nnf_bilinear {torch}R Documentation

Bilinear

Description

Applies a bilinear transformation to the incoming data: y=x1Ax2+by = x_1 A x_2 + b

Usage

nnf_bilinear(input1, input2, weight, bias = NULL)

Arguments

input1

(N,,Hin1)(N, *, H_{in1}) where Hin1=\mboxin1_featuresH_{in1}=\mbox{in1\_features} and * means any number of additional dimensions. All but the last dimension of the inputs should be the same.

input2

(N,,Hin2)(N, *, H_{in2}) where Hin2=\mboxin2_featuresH_{in2}=\mbox{in2\_features}

weight

(\mboxout_features,\mboxin1_features,\mboxin2_features)(\mbox{out\_features}, \mbox{in1\_features}, \mbox{in2\_features})

bias

(\mboxout_features)(\mbox{out\_features})

Value

output (N,,Hout)(N, *, H_{out}) where Hout=\mboxout_featuresH_{out}=\mbox{out\_features} and all but the last dimension are the same shape as the input.


[Package torch version 0.13.0 Index]