Skip to content

Commit 49074d9

Browse files
committed
Fixed final median shape issue
1 parent ff8b026 commit 49074d9

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
@@ -1192,6 +1192,8 @@ def median(x, axis=None, keepdims=False):
11921192
).output(0)
11931193
else:
11941194
median_eval = ov_opset.unsqueeze(median_eval, ov_axis).output(0)
1195+
else:
1196+
median_eval = ov_opset.squeeze(median_eval, ov_axis_positive).output(0)
11951197

11961198
return OpenVINOKerasTensor(median_eval)
11971199

0 commit comments

Comments
 (0)