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 {
7777 AddOutput (" Out" , " The output of bilinear_tensor_product operator." );
7878 AddComment (R"DOC(
7979Bilinear 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$$;
8593
8694)DOC" );
8795 }
You can’t perform that action at this time.
0 commit comments