File tree Expand file tree Collapse file tree 1 file changed +13
-5
lines changed Expand file tree Collapse file tree 1 file changed +13
-5
lines changed Original file line number Diff line number Diff line change @@ -77,11 +77,19 @@ class BilinearTensorProductOpMaker : public framework::OpProtoAndCheckerMaker {
77
77
AddOutput (" Out" , " The output of bilinear_tensor_product operator." );
78
78
AddComment (R"DOC(
79
79
Bilinear Tensor Product operator.
80
- Given input X and Y, a 3D tensor weight, and bias. Each column of the
81
- output is computed by one slice i = 1, . . . , k of the tensor:
82
-
83
- M = (X W_i) \cdot Y
84
- Out_i = \sum_i {M_i} + Bias_i
80
+ Given input X and Y, a 3D tensor Weight and a Bias. Each column of the
81
+ Output is computed by one slice $i = 1, . . . , k$ of the tensor:
82
+
83
+ $$
84
+ M = (X W_i) * Y \\
85
+ Out_i = \sum_j {M_j} + Bias_i
86
+ $$
87
+
88
+ Where $W_i$ is the $i$-th slice of Input(Weight);
89
+ $M_j$ is the $j$-th column of $M$;
90
+ $Out_i$ is the $i$-th column of Output(Out);
91
+ $Bias_i$ is a column vector, each element of it is equal to
92
+ the $i$-th element of $Bias$;
85
93
86
94
)DOC" );
87
95
}
You can’t perform that action at this time.
0 commit comments