File tree Expand file tree Collapse file tree 1 file changed +14
-6
lines changed Expand file tree Collapse file tree 1 file changed +14
-6
lines changed Original file line number Diff line number Diff line change @@ -60,12 +60,16 @@ class WriteToArrayOpProtoMaker : public framework::OpProtoAndCheckerMaker {
60
60
" (Tensor) the subscript index in tensor array. The number of element "
61
61
" should be 1" );
62
62
AddOutput (" Out" , " (TensorArray) the tensor array will be written" );
63
- AddComment (R"DOC( Write a LoDTensor to a LoDTensor array.
63
+ AddComment (R"DOC(
64
+ WriteToArray Operator.
64
65
65
- Assume T is LoDTensor, i is the subscript of the array, and A is the array. The
66
+ This operator writes a LoDTensor to a LoDTensor array.
67
+
68
+ Assume $T$ is LoDTensor, $i$ is the subscript of the array, and $A$ is the array. The
66
69
equation is
67
70
68
- A[i] = T
71
+ $$A[i] = T$$
72
+
69
73
)DOC" );
70
74
}
71
75
};
@@ -144,12 +148,16 @@ class ReadFromArrayProtoMaker : public framework::OpProtoAndCheckerMaker {
144
148
" (Tensor) the subscript index in tensor array. The number of "
145
149
" element should be 1" );
146
150
AddOutput (" Out" , " (LoDTensor) the tensor will be read from." );
147
- AddComment (R"DOC( Read a LoDTensor from a LoDTensor Array
151
+ AddComment (R"DOC(
152
+ ReadFromArray Operator.
148
153
149
- Assume T is LoDTensor, i is th e subscript of the array, and A is the array. The
154
+ Read a LoDTensor from a LoDTensor Array.
155
+
156
+ Assume $T$ is LoDTensor, $i$ is the subscript of the array, and $A$ is the array. The
150
157
equation is
151
158
152
- T = A[i]
159
+ $$T = A[i]$$
160
+
153
161
)DOC" );
154
162
}
155
163
};
You can’t perform that action at this time.
0 commit comments