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