Skip to content

Commit 734e05f

Browse files
committed
Fixed squeeze missing axis
1 parent 9f877b7 commit 734e05f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras/src/backend/openvino/numpy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1147,7 +1147,7 @@ def median(x, axis=None, keepdims=False):
11471147
else:
11481148
ov_axis_positive = ov_axis
11491149

1150-
k_scalar = ov_opset.squeeze(k_value).output(0)
1150+
k_scalar = ov_opset.squeeze(k_value, ov_opset.constant([0], Type.i32).output(0)).output(0)
11511151
x_sorted = ov_opset.topk(
11521152
x, k_scalar, axis, "min", "value", stable=True
11531153
).output(0)

0 commit comments

Comments
 (0)