Skip to content

Commit 0e187bc

Browse files
chengduoabhinavarora
authored andcommitted
fix get_mid_dims annotation (#8490)
1 parent 77ee8fb commit 0e187bc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/operators/elementwise_op_function.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,10 @@ namespace operators {
3535
* For example:
3636
* 1. shape(X) = (2, 3, 4, 5), shape(Y) = (3, 4), with axis=1
3737
* pre=2, n=3*4, post=5
38-
* x.shape(2, 12, 5) * y.shape(1,12,1).broadcast(2,12,5)
38+
* x.shape(2, 12, 5) * y.shape(1, 12, 1).broadcast(2, 12, 5)
3939
* 2. shape(X) = (2, 3, 4, 5), shape(Y) = (4,5)
4040
* pre=2*3, n=4*5, post=1
41-
* x.shape(2, 3, 20) * y.shape(1,1,20).broadcast(2,3,20)
41+
* x.shape(6, 20, 1) * y.shape(1, 20, 1).broadcast(6, 20, 1)
4242
*/
4343
inline void get_mid_dims(const framework::DDim& x_dims,
4444
const framework::DDim& y_dims, const int axis,

0 commit comments

Comments
 (0)