Fix broadcast ops descriptions (#21087)

apache · Jul 7, 2022 · 26243ee · 26243ee
1 parent 84b1626
commit 26243ee
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/operator/tensor/elemwise_binary_broadcast_op_extended.cc b/src/operator/tensor/elemwise_binary_broadcast_op_extended.cc
@@ -39,8 +39,8 @@ Example::
  y = [[ 0.],
  [ 1.]]
 
- broadcast_power(x, y) = [[ 2., 2., 2.],
- [ 4., 4., 4.]]
+ broadcast_power(x, y) = [[ 1., 1., 1.],
+ [ 1., 1., 1.]]
 
 )code" ADD_FILELINE)
  .set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::power>)
@@ -112,7 +112,7 @@ Example::
  y = [[ 0.],
  [ 1.]]
 
- broadcast_maximum(x, y) = [[ 0., 0., 0.],
+ broadcast_minimum(x, y) = [[ 0., 0., 0.],
  [ 1., 1., 1.]]
 
 )code" ADD_FILELINE)