|
|
|
@ -45,33 +45,36 @@ class SequencePoolOpMaker : public framework::OpProtoAndCheckerMaker {
|
|
|
|
|
.SetDefault("AVERAGE")
|
|
|
|
|
.InEnum({"AVERAGE", "SUM", "SQRT", "LAST", "FIRST", "MAX"});
|
|
|
|
|
AddComment(R"DOC(
|
|
|
|
|
SequencePoolOp pools features of all time-steps of each instance.
|
|
|
|
|
|
|
|
|
|
It supports six pooling pooltype:
|
|
|
|
|
- AVERAGE: Out[i] = average_{for each instance in i-th sequence}{X[i]}
|
|
|
|
|
- SUM: Out[i] = sum_{for each instance in i-th sequence}{X[i]}
|
|
|
|
|
- SQRT: Out[i] = sum_{for each instance in i-th sequence}{X[i]}
|
|
|
|
|
/ sqrt(i-th sequence length)
|
|
|
|
|
- LAST: Out[i] = last instance in i-th sequence X[i]
|
|
|
|
|
- FIRST: Out[i] = first instance in i-th sequence X[i]
|
|
|
|
|
- MAX: Out[i] = max_{for each instance in i-th sequence}{X[i]}
|
|
|
|
|
|
|
|
|
|
For a mini-batch of 3 variable-length sentences, containing 2, 3, and 2 time-steps:
|
|
|
|
|
|
|
|
|
|
Assume X is a [7,M,N] LoDTensor, and X->lod()[0] = [0, 2, 5, 7], 7=2+3+2.
|
|
|
|
|
Besides, for the sake of simplicity, we assume M=1 and N=1,
|
|
|
|
|
and the value of X = [[1, 3], [2, 4, 6], [5, 1]].
|
|
|
|
|
|
|
|
|
|
Thus, Out is a [3,1,1] Tensor without LoD infomation.
|
|
|
|
|
And for different pooltype, the value of Out is as follows:
|
|
|
|
|
|
|
|
|
|
- AVERAGE: [2, 4, 3], where 2=(1+3)/2, 4=(2+4+6)/3, 3=(5+1)/2
|
|
|
|
|
- SUM: [4, 12, 6], where 4=1+3, 12=2+4+6, 6=5+1
|
|
|
|
|
- SQRT: [2.82, 6.93, 4.24], where 2.82=(1+3)/sqrt(2),
|
|
|
|
|
Sequence Pool Operator.
|
|
|
|
|
|
|
|
|
|
The SequencePoolOp pools features of all time-steps of each instance.
|
|
|
|
|
It supports six pooling types:
|
|
|
|
|
1. AVERAGE: Out[i] = $$avg(X_i)$$
|
|
|
|
|
2. SUM: Out[i] = $$\sum_jX_{ij}$$
|
|
|
|
|
3. SQRT: Out[i] = $$\frac{\sum_jX_{ij}}{\sqrt{len(X_i)}}$$
|
|
|
|
|
4. LAST: Out[i] = last instance in i-th sequence X[i]
|
|
|
|
|
5. FIRST: Out[i] = first instance in i-th sequence X[i]
|
|
|
|
|
6. MAX: Out[i] = $$max(X_i)$$
|
|
|
|
|
|
|
|
|
|
The following example explains how this works:
|
|
|
|
|
For a mini-batch of 3 variable-length sentences,
|
|
|
|
|
containing 2, 3, and 2 time-steps:
|
|
|
|
|
|
|
|
|
|
Assume X is a [7,M,N] LoDTensor, and X->lod()[0] = [0, 2, 5, 7], 7=2+3+2.
|
|
|
|
|
Besides, for the sake of simplicity, we assume M=1 and N=1,
|
|
|
|
|
and the value of X = [[1, 3], [2, 4, 6], [5, 1]].
|
|
|
|
|
|
|
|
|
|
Thus, Out is a [3,1,1] Tensor without LoD infomation.
|
|
|
|
|
And for different pooltype, the value of Out is as follows:
|
|
|
|
|
|
|
|
|
|
- AVERAGE: [2, 4, 3], where 2=(1+3)/2, 4=(2+4+6)/3, 3=(5+1)/2
|
|
|
|
|
- SUM: [4, 12, 6], where 4=1+3, 12=2+4+6, 6=5+1
|
|
|
|
|
- SQRT: [2.82, 6.93, 4.24], where 2.82=(1+3)/sqrt(2),
|
|
|
|
|
6.93=(2+4+6)/sqrt(3), 4.24=(5+1)/sqrt(2)
|
|
|
|
|
- MAX: [3, 6, 5], where 3=max(1,3), 6=max(2,4,6), 5=max(5,1)
|
|
|
|
|
- LAST: [3, 6, 1], where 3=last(1,3), 6=last(2,4,6), 1=last(5,1)
|
|
|
|
|
- FIRST: [1, 2, 5], where 1=first(1,3), 2=first(2,4,6), 5=first(5,1)
|
|
|
|
|
- MAX: [3, 6, 5], where 3=max(1,3), 6=max(2,4,6), 5=max(5,1)
|
|
|
|
|
- LAST: [3, 6, 1], where 3=last(1,3), 6=last(2,4,6), 1=last(5,1)
|
|
|
|
|
- FIRST: [1, 2, 5], where 1=first(1,3), 2=first(2,4,6), 5=first(5,1)
|
|
|
|
|
|
|
|
|
|
)DOC");
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|