Skip to content

Commit 0eeaa5e

Browse files
committed
added comment to median function
1 parent 04e72cf commit 0eeaa5e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

keras/src/backend/openvino/numpy.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1150,6 +1150,8 @@ def median(x, axis=None, keepdims=False):
11501150
x_rank_original,
11511151
ov_opset.constant([len(axis) - 1], Type.i32).output(0),
11521152
).output(0)
1153+
# create flatten shape of 0's (keep axes)
1154+
# and -1 at the end (flattened axis)
11531155
x_flatten_shape = ov_opset.broadcast(
11541156
ov_opset.constant([0], Type.i32).output(0), x_flatten_rank
11551157
).output(0)

0 commit comments

Comments
 (0)