From 46dae2f7f2de4f3601dc1e82750cf14ab6d050a6 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 17 Oct 2024 16:34:06 -0400 Subject: [PATCH 01/11] Update python example and benchmark it --- benches/example_benchmarks.rs | 3 +- tests/python_array_optimize.egg | 1664 +++++++++++++++++++++++++++---- 2 files changed, 1456 insertions(+), 211 deletions(-) diff --git a/benches/example_benchmarks.rs b/benches/example_benchmarks.rs index 9ca1e81f..e733986a 100644 --- a/benches/example_benchmarks.rs +++ b/benches/example_benchmarks.rs @@ -11,8 +11,7 @@ pub fn criterion_benchmark(c: &mut Criterion) { for entry in glob::glob("tests/**/*.egg").unwrap() { let path = entry.unwrap().clone(); let path_string = path.to_string_lossy().to_string(); - // Skip python_array_optimize since it is too slow and doesn't even reflect the current python implementation - if path_string.contains("fail-typecheck") || path_string.contains("python_array_optimize") { + if path_string.contains("fail-typecheck") { continue; } let name = path.file_stem().unwrap().to_string_lossy().to_string(); diff --git a/tests/python_array_optimize.egg b/tests/python_array_optimize.egg index 7be04f5d..a24745c2 100644 --- a/tests/python_array_optimize.egg +++ b/tests/python_array_optimize.egg @@ -1,48 +1,676 @@ +;; created by running test_source_optimized and removing the PyObject functions +;; pytest -s python/tests/test_array_api.py::TestLDA::test_source_optimized --benchmark-disable +(push 1) +(push 1) +(sort DType) +(function DType_float64 () DType) +(let %__expr_4101187900440652363 (DType_float64)) +(sort Boolean) +(function DType___eq__ (DType DType) Boolean) +(sort NDArray) +(function NDArray_dtype (NDArray) DType) +(function assume_isfinite (NDArray) NDArray) +(sort TupleInt) +(function assume_shape (NDArray TupleInt) NDArray) +(function assume_dtype (NDArray DType) NDArray) +(function NDArray_var (String) NDArray :cost 200) +(sort Int) +(sort Vec_Int (Vec Int)) +(function TupleInt_from_vec (Vec_Int) TupleInt) +(function Int___init__ (i64) Int) +(DType___eq__ %__expr_4101187900440652363 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) +(ruleset array_api_ruleset) +(function i-Int__-Int_i (Int Int) Int) +(rewrite (i-Int__-Int_i i _) i :ruleset array_api_ruleset) +(function TupleInt_single (Int) TupleInt) +(sort UnstableFn_Int_Int (UnstableFn (Int) Int)) +(function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt :unextractable) +(rewrite (TupleInt_single i) (TupleInt___init__ (Int___init__ 1) (unstable-fn "i-Int__-Int_i" i)) :ruleset array_api_ruleset) +(function i-Int_i (Int) Int) +(rewrite (i-Int_i i) i :ruleset array_api_ruleset) +(function TupleInt_range (Int) TupleInt) +(rewrite (TupleInt_range stop) (TupleInt___init__ stop (unstable-fn "i-Int_i")) :ruleset array_api_ruleset) +(function index_vec_int (Vec_Int Int) Int) +(rewrite (TupleInt_from_vec vec) (TupleInt___init__ (Int___init__ (vec-length vec)) (unstable-fn "index_vec_int" vec)) :ruleset array_api_ruleset) +(function Int_if_ (Boolean Int Int) Int) +(function Int___lt__ (Int Int) Boolean) +(function TupleInt_length (TupleInt) Int) +(function TupleInt___getitem__ (TupleInt Int) Int) +(function Int___sub__ (Int Int) Int) +(function other-TupleInt_self-TupleInt_i-Int__Int_if___Int___lt___i__TupleInt_length_self____TupleInt___getitem___self_i___TupleInt___getitem___other__Int___sub___i__TupleInt_length_self____ (TupleInt TupleInt Int) Int) +(rewrite (other-TupleInt_self-TupleInt_i-Int__Int_if___Int___lt___i__TupleInt_length_self____TupleInt___getitem___self_i___TupleInt___getitem___other__Int___sub___i__TupleInt_length_self____ other self i) (Int_if_ (Int___lt__ i (TupleInt_length self)) (TupleInt___getitem__ self i) (TupleInt___getitem__ other (Int___sub__ i (TupleInt_length self)))) :ruleset array_api_ruleset) +(function TupleInt___add__ (TupleInt TupleInt) TupleInt) +(function Int___add__ (Int Int) Int) +(rewrite (TupleInt___add__ self other) (TupleInt___init__ (Int___add__ (TupleInt_length self) (TupleInt_length other)) (unstable-fn "other-TupleInt_self-TupleInt_i-Int__Int_if___Int___lt___i__TupleInt_length_self____TupleInt___getitem___self_i___TupleInt___getitem___other__Int___sub___i__TupleInt_length_self____" other self)) :ruleset array_api_ruleset) +(function Boolean___or__ (Boolean Boolean) Boolean) +(function Int___eq__ (Int Int) Boolean) +(function i-Int_acc-Boolean_j-Int__Boolean___or___acc__Int___eq___i_j__ (Int Boolean Int) Boolean) +(rewrite (i-Int_acc-Boolean_j-Int__Boolean___or___acc__Int___eq___i_j__ i acc j) (Boolean___or__ acc (Int___eq__ i j)) :ruleset array_api_ruleset) +(function TupleInt_contains (TupleInt Int) Boolean) +(sort UnstableFn_Boolean_Boolean_Int (UnstableFn (Boolean Int) Boolean)) +(function TupleInt_fold_boolean (TupleInt Boolean UnstableFn_Boolean_Boolean_Int) Boolean) +(function FALSE () Boolean) +(rewrite (TupleInt_contains self i) (TupleInt_fold_boolean self (FALSE) (unstable-fn "i-Int_acc-Boolean_j-Int__Boolean___or___acc__Int___eq___i_j__" i)) :ruleset array_api_ruleset) +(function f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__ (UnstableFn_Int_Int TupleInt Int) Int) +(rewrite (f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__ f self i) (unstable-app f (TupleInt___getitem__ self i)) :ruleset array_api_ruleset) +(function TupleInt_map (TupleInt UnstableFn_Int_Int) TupleInt :cost 100) +(rewrite (TupleInt_map self f) (TupleInt___init__ (TupleInt_length self) (unstable-fn "f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__" f self)) :ruleset array_api_ruleset) +(function i-TupleInt__-Int_i (TupleInt Int) TupleInt) +(rewrite (i-TupleInt__-Int_i i _) i :ruleset array_api_ruleset) +(sort TupleTupleInt) +(function TupleTupleInt_single (TupleInt) TupleTupleInt) +(sort UnstableFn_TupleInt_Int (UnstableFn (Int) TupleInt)) +(function TupleTupleInt___init__ (Int UnstableFn_TupleInt_Int) TupleTupleInt) +(rewrite (TupleTupleInt_single i) (TupleTupleInt___init__ (Int___init__ 1) (unstable-fn "i-TupleInt__-Int_i" i)) :ruleset array_api_ruleset) +(function TupleTupleInt_from_vec (Vec_Int) TupleInt) +(rewrite (TupleTupleInt_from_vec vec) (TupleInt___init__ (Int___init__ (vec-length vec)) (unstable-fn "index_vec_int" vec)) :ruleset array_api_ruleset) +(function TupleInt_if_ (Boolean TupleInt TupleInt) TupleInt) +(function TupleTupleInt_length (TupleTupleInt) Int) +(function TupleTupleInt___getitem__ (TupleTupleInt Int) TupleInt) +(function other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ (TupleTupleInt TupleTupleInt Int) TupleInt) +(rewrite (other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ other self i) (TupleInt_if_ (Int___lt__ i (TupleTupleInt_length self)) (TupleTupleInt___getitem__ self i) (TupleTupleInt___getitem__ other (Int___sub__ i (TupleTupleInt_length self)))) :ruleset array_api_ruleset) +(function TupleTupleInt___add__ (TupleTupleInt TupleTupleInt) TupleTupleInt) +(rewrite (TupleTupleInt___add__ self other) (TupleTupleInt___init__ (Int___add__ (TupleTupleInt_length self) (TupleTupleInt_length other)) (unstable-fn "other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____" other self)) :ruleset array_api_ruleset) +(function NDArray_size (NDArray) Int) +(sort UnstableFn_Int_Int_Int (UnstableFn (Int Int) Int)) +(function TupleInt_fold (TupleInt Int UnstableFn_Int_Int_Int) Int) +(function NDArray_shape (NDArray) TupleInt) +(function Int___mul__ (Int Int) Int) +(rewrite (NDArray_size x) (TupleInt_fold (NDArray_shape x) (Int___init__ 1) (unstable-fn "Int___mul__")) :ruleset array_api_ruleset) +(function unique_values (NDArray) NDArray) +(sort TupleValue) +(function NDArray_vector (TupleValue) NDArray) +(sort Value) +(function possible_values (Value) TupleValue) +(function NDArray_index (NDArray TupleInt) Value) +(function ALL_INDICES () TupleInt) +(rewrite (unique_values a) (NDArray_vector (possible_values (NDArray_index a (ALL_INDICES)))) :ruleset array_api_ruleset) +(function Value_isfinite (Value) Boolean) +(function Value_int (Int) Value) +(function TRUE () Boolean) +(rewrite (Value_isfinite (Value_int i)) (TRUE) :ruleset array_api_ruleset) +(function Value_bool (Boolean) Value) +(rewrite (Value_isfinite (Value_bool b)) (TRUE) :ruleset array_api_ruleset) +(sort Float) +(function Value_float (Float) Value) +(function Float___init__ (f64) Float :cost 3) +(rewrite (Value_isfinite (Value_float (Float___init__ f))) (TRUE) :when ((!= f NaN)) :ruleset array_api_ruleset) +(function isfinite (NDArray) NDArray) +(sort OptionalIntOrTuple) +(function sum (NDArray OptionalIntOrTuple) NDArray) +(function OptionalIntOrTuple_none () OptionalIntOrTuple) +(function NDArray_scalar (Value) NDArray) +(rewrite (isfinite (sum arr (OptionalIntOrTuple_none))) (NDArray_scalar (Value_bool (Value_isfinite (NDArray_index arr (ALL_INDICES))))) :ruleset array_api_ruleset) +(function assume_value_one_of (NDArray TupleValue) NDArray) +(rewrite (NDArray_shape (assume_value_one_of x vs)) (NDArray_shape x) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_value_one_of x vs)) (NDArray_dtype x) :ruleset array_api_ruleset) +(rule ((= v (NDArray_index (assume_value_one_of x vs) idx))) + ((union v (NDArray_index x idx)) + (union (possible_values v) vs)) + :ruleset array_api_ruleset ) +(rewrite (NDArray_shape (assume_isfinite x)) (NDArray_shape x) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_isfinite x)) (NDArray_dtype x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_isfinite x) ti) (NDArray_index x ti) :ruleset array_api_ruleset) +(rewrite (Value_isfinite (NDArray_index (assume_isfinite x) ti)) (TRUE) :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_shape x shape)) shape :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_shape x shape)) (NDArray_dtype x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_shape x shape) idx) (NDArray_index x idx) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_dtype x dtype)) dtype :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_dtype x dtype)) (NDArray_shape x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_dtype x dtype) idx) (NDArray_index x idx) :ruleset array_api_ruleset) +(sort IndexKey) +(function NDArray___getitem__ (NDArray IndexKey) NDArray) +(function IndexKey_int (Int) IndexKey) +(rewrite (NDArray___getitem__ x (IndexKey_int i)) (NDArray_scalar (NDArray_index x (TupleInt_single i))) :ruleset array_api_ruleset) +(sort OptionalBool) +(function reshape (NDArray TupleInt OptionalBool) NDArray) +(rule ((= __a (reshape x shape copy))) + ((NDArray_shape x) + (TupleInt_length (NDArray_shape x))) + :ruleset array_api_ruleset ) +(rule ((reshape x shape copy)) + ((TupleInt_length shape) + (TupleInt___getitem__ shape (Int___init__ 0))) + :ruleset array_api_ruleset ) +(rewrite (reshape x shape copy) x :when ((= (TupleInt_length (NDArray_shape x)) (Int___init__ 1)) (= (TupleInt_length shape) (Int___init__ 1)) (= (TupleInt___getitem__ shape (Int___init__ 0)) (Int___init__ -1))) :ruleset array_api_ruleset) +(function TupleValue_length (TupleValue) Int) +(rewrite (NDArray_shape (NDArray_vector vs)) (TupleInt_single (TupleValue_length vs)) :ruleset array_api_ruleset) +(function Value_dtype (Value) DType) +(function TupleValue___getitem__ (TupleValue Int) Value) +(rewrite (NDArray_dtype (NDArray_vector vs)) (Value_dtype (TupleValue___getitem__ vs (Int___init__ 0))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_vector vs) ti) (TupleValue___getitem__ vs (TupleInt___getitem__ ti (Int___init__ 0))) :ruleset array_api_ruleset) +(function TupleInt_EMPTY () TupleInt) +(rewrite (NDArray_shape (NDArray_scalar v)) (TupleInt_EMPTY) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (NDArray_scalar v)) (Value_dtype v) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_scalar v) (TupleInt_EMPTY)) v :ruleset array_api_ruleset) +(function any (NDArray) NDArray) +(function TupleValue_includes (TupleValue Value) Boolean) +(function Value_to_truthy_value (Value) Value) +(rewrite (any x) (NDArray_scalar (Value_bool (TupleValue_includes (possible_values (Value_to_truthy_value (NDArray_index x (ALL_INDICES)))) (Value_bool (TRUE))))) :ruleset array_api_ruleset) +(function NDArray___lt__ (NDArray NDArray) NDArray) +(function Value___lt__ (Value Value) Value) +(function broadcast_index (TupleInt TupleInt TupleInt) TupleInt) +(function broadcast_shapes (TupleInt TupleInt) TupleInt) +(rewrite (NDArray_index (NDArray___lt__ x y) idx) (Value___lt__ (NDArray_index x (broadcast_index (NDArray_shape x) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx)) (NDArray_index y (broadcast_index (NDArray_shape y) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx))) :ruleset array_api_ruleset) +(function NDArray___truediv__ (NDArray NDArray) NDArray) +(function Value___truediv__ (Value Value) Value) +(rewrite (NDArray_index (NDArray___truediv__ x y) idx) (Value___truediv__ (NDArray_index x (broadcast_index (NDArray_shape x) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx)) (NDArray_index y (broadcast_index (NDArray_shape y) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_scalar v) idx) v :ruleset array_api_ruleset) +(function astype (NDArray DType) NDArray) +(function Value_astype (Value DType) Value) +(rewrite (NDArray_index (astype x dtype) idx) (Value_astype (NDArray_index x idx) dtype) :ruleset array_api_ruleset) +(relation greater_zero (Value)) +(sort TupleNDArray) +(function TupleNDArray___getitem__ (TupleNDArray Int) NDArray) +(function unique_counts (NDArray) TupleNDArray) +(rule ((= v (NDArray_index (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) idx))) + ((greater_zero v)) + :ruleset array_api_ruleset ) +(rule ((greater_zero v) + (= v1 (Value_astype v dtype))) + ((greater_zero v1)) + :ruleset array_api_ruleset ) +(rule ((= v (Value_float (Float___init__ f))) + (> f 0.0)) + ((greater_zero v)) + :ruleset array_api_ruleset ) +(rule ((= v (Value_int (Int___init__ i))) + (> i 0)) + ((greater_zero v)) + :ruleset array_api_ruleset ) +(rule ((greater_zero v) + (greater_zero v1) + (= v2 (Value___truediv__ v v1))) + ((greater_zero v2)) + :ruleset array_api_ruleset ) +(rule ((greater_zero v) + (= v1 (Value___lt__ v (Value_int (Int___init__ 0))))) + ((union v1 (Value_bool (FALSE)))) + :ruleset array_api_ruleset ) +(function TupleValue___init__ (Value) TupleValue) +(rewrite (possible_values (Value_bool b)) (TupleValue___init__ (Value_bool b)) :ruleset array_api_ruleset) +(rule ((= v1 (Value_astype v dtype)) + (greater_zero v)) + ((greater_zero v1)) + :ruleset array_api_ruleset ) +(function TupleNDArray_length (TupleNDArray) Int) +(function svd (NDArray Boolean) TupleNDArray) +(rewrite (TupleNDArray_length (svd x full_matrices)) (Int___init__ 3) :ruleset array_api_ruleset) +(function unique_inverse (NDArray) TupleNDArray) +(rewrite (TupleNDArray_length (unique_inverse x)) (Int___init__ 2) :ruleset array_api_ruleset) +(rewrite (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 0)) (unique_values x) :ruleset array_api_ruleset) +(function ndarray-abs (NDArray) NDArray) +(function Float_abs (Float) Float) +(rewrite (ndarray-abs (NDArray_scalar (Value_float f))) (NDArray_scalar (Value_float (Float_abs f))) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (unique_counts x)) (Int___init__ 2) :ruleset array_api_ruleset) +(rewrite (sum (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_int (NDArray_size x))) :ruleset array_api_ruleset) +(rewrite (sum (astype (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) dtype) (OptionalIntOrTuple_none)) (astype (NDArray_scalar (Value_int (NDArray_size x))) dtype) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (astype x dtype)) dtype :ruleset array_api_ruleset) +(rewrite (astype (NDArray_scalar (Value_int (Int___init__ i))) %__expr_4101187900440652363) (NDArray_scalar (Value_float (Float___init__ (to-f64 i)))) :ruleset array_api_ruleset) +(sort OptionalInt) +(function concat (TupleNDArray OptionalInt) NDArray) +(function TupleNDArray___init__ (NDArray) TupleNDArray) +(function OptionalInt_none () OptionalInt) +(rewrite (concat (TupleNDArray___init__ x) (OptionalInt_none)) x :ruleset array_api_ruleset) +(rewrite (unique_values (unique_values x)) (unique_values x) :ruleset array_api_ruleset) +(rewrite (sum (NDArray___truediv__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_none)) (NDArray_scalar v)) :ruleset array_api_ruleset) +(function NDArray_ndim (NDArray) Int) +(sort OptionalDType) +(sort OptionalDevice) +(function asarray (NDArray OptionalDType OptionalBool OptionalDevice) NDArray) +(function OptionalDevice_none () OptionalDevice) +(rewrite (NDArray_ndim (asarray a d ob (OptionalDevice_none))) (NDArray_ndim a) :ruleset array_api_ruleset) +(function OptionalDType_none () OptionalDType) +(function OptionalBool_none () OptionalBool) +(rewrite (asarray a (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) a :ruleset array_api_ruleset) +(function TupleNDArray___add__ (TupleNDArray TupleNDArray) TupleNDArray) +(function TupleNDArray_EMPTY () TupleNDArray) +(rewrite (TupleNDArray___add__ ti (TupleNDArray_EMPTY)) ti :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (TupleNDArray___init__ n)) (Int___init__ 1) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (TupleNDArray___add__ ti ti2)) (Int___add__ (TupleNDArray_length ti) (TupleNDArray_length ti2)) :ruleset array_api_ruleset) +(sort UnstableFn_Value_TupleInt (UnstableFn (TupleInt) Value)) +(function NDArray___init__ (TupleInt DType UnstableFn_Value_TupleInt) NDArray) +(rewrite (NDArray_shape (NDArray___init__ shape dtype idx_fn)) shape :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (NDArray___init__ shape dtype idx_fn)) dtype :ruleset array_api_ruleset) +(rewrite (NDArray_ndim x) (TupleInt_length (NDArray_shape x)) :ruleset array_api_ruleset) +(function NDArray_to_value (NDArray) Value) +(rewrite (NDArray_to_value x) (NDArray_index x (TupleInt_EMPTY)) :ruleset array_api_ruleset) +(rewrite (NDArray___truediv__ (NDArray_scalar (Value_float f)) (NDArray_scalar (Value_float f))) (NDArray_scalar (Value_float (Float___init__ 1.0))) :ruleset array_api_ruleset) +(function NDArray___sub__ (NDArray NDArray) NDArray) +(rewrite (NDArray___sub__ (NDArray_scalar (Value_float f)) (NDArray_scalar (Value_float f))) (NDArray_scalar (Value_float (Float___init__ 0.0))) :ruleset array_api_ruleset) +(function NDArray___gt__ (NDArray NDArray) NDArray) +(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ fi1))) (NDArray_scalar (Value_float (Float___init__ fi2)))) (NDArray_scalar (Value_bool (TRUE))) :when ((> fi1 fi2)) :ruleset array_api_ruleset) +(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ fi1))) (NDArray_scalar (Value_float (Float___init__ fi2)))) (NDArray_scalar (Value_bool (FALSE))) :when ((<= fi1 fi2)) :ruleset array_api_ruleset) +(function NDArray_T (NDArray) NDArray) +(rewrite (NDArray_T (NDArray_T x)) x :ruleset array_api_ruleset) +(function TupleValue___add__ (TupleValue TupleValue) TupleValue) +(function TupleValue_EMPTY () TupleValue) +(rewrite (TupleValue___add__ ti (TupleValue_EMPTY)) ti :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue_EMPTY)) (Int___init__ 0) :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue___init__ v)) (Int___init__ 1) :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue___add__ ti ti2)) (Int___add__ (TupleValue_length ti) (TupleValue_length ti2)) :ruleset array_api_ruleset) +(rewrite (TupleValue___getitem__ (TupleValue___init__ v) (Int___init__ 0)) v :ruleset array_api_ruleset) +(rewrite (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ v) ti) (Int___init__ 0)) v :ruleset array_api_ruleset) +(rule ((= v (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ v2) ti) (Int___init__ k))) + (> k 0)) + ((union v (TupleValue___getitem__ ti (Int___init__ (- k 1))))) + :ruleset array_api_ruleset ) +(rewrite (TupleValue_includes (TupleValue_EMPTY) v) (FALSE) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___init__ v) v) (TRUE) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___init__ v) v2) (FALSE) :when ((!= v v2)) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___add__ ti ti2) v) (Boolean___or__ (TupleValue_includes ti v) (TupleValue_includes ti2 v)) :ruleset array_api_ruleset) +(function DType_int64 () DType) +(rewrite (Value_dtype (Value_int i)) (DType_int64) :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_float f)) %__expr_4101187900440652363 :ruleset array_api_ruleset) +(function DType_bool () DType) +(rewrite (Value_dtype (Value_bool b)) (DType_bool) :ruleset array_api_ruleset) +(function Value_to_bool (Value) Boolean) +(rewrite (Value_to_bool (Value_bool b)) b :ruleset array_api_ruleset) +(function Value_to_int (Value) Int) +(rewrite (Value_to_int (Value_int i)) i :ruleset array_api_ruleset) +(rewrite (Value_to_truthy_value (Value_bool b)) (Value_bool b) :ruleset array_api_ruleset) +(sort IsDtypeKind) +(function isdtype (DType IsDtypeKind) Boolean) +(function DType_float32 () DType) +(function IsDtypeKind_string (String) IsDtypeKind) +(rewrite (isdtype (DType_float32) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_4101187900440652363 (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(function DType_object () DType) +(rewrite (isdtype (DType_object) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int64) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) +(function DType_int32 () DType) +(rewrite (isdtype (DType_int32) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float32) (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_4101187900440652363 (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_object) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int64) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int32) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_4101187900440652363 (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_object) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int64) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(function IsDtypeKind_NULL () IsDtypeKind) +(rewrite (isdtype d (IsDtypeKind_NULL)) (FALSE) :ruleset array_api_ruleset) +(function IsDtypeKind_dtype (DType) IsDtypeKind) +(rewrite (isdtype d (IsDtypeKind_dtype d)) (TRUE) :ruleset array_api_ruleset) +(function IsDtypeKind___or__ (IsDtypeKind IsDtypeKind) IsDtypeKind :cost 10) +(rewrite (isdtype d (IsDtypeKind___or__ k1 k2)) (Boolean___or__ (isdtype d k1) (isdtype d k2)) :ruleset array_api_ruleset) +(rewrite (IsDtypeKind___or__ k1 (IsDtypeKind_NULL)) k1 :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_4101187900440652363 %__expr_4101187900440652363) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_float32)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_int32)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_int64)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_object)) (TRUE) :ruleset array_api_ruleset) +(function idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ (UnstableFn_Int_Int Int) Int) +(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ idx_fn i) (unstable-app idx_fn (Int___add__ i (Int___init__ 1))) :ruleset array_api_ruleset) +(rewrite (TupleInt_length (TupleInt___init__ i idx_fn)) i :ruleset array_api_ruleset) +(rewrite (TupleInt___getitem__ (TupleInt___init__ i idx_fn) i2) (unstable-app idx_fn i2) :ruleset array_api_ruleset) +(rewrite (index_vec_int vs (Int___init__ k)) (vec-get vs k) :when ((> (vec-length vs) k)) :ruleset array_api_ruleset) +(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ 0) idx_fn) i f) i :ruleset array_api_ruleset) +(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ k) idx_fn) i f) (unstable-app f (TupleInt_fold (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) i f) (unstable-app idx_fn (Int___init__ 0))) :when ((!= k 0)) :ruleset array_api_ruleset) +(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ 0) idx_fn) b bool_f) b :ruleset array_api_ruleset) +(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ k) idx_fn) b bool_f) (unstable-app bool_f (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) b bool_f) (unstable-app idx_fn (Int___init__ 0))) :when ((!= k 0)) :ruleset array_api_ruleset) +(sort UnstableFn_Boolean_Int (UnstableFn (Int) Boolean)) +(function TupleInt_filter (TupleInt UnstableFn_Boolean_Int) TupleInt :cost 100) +(rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ 0) idx_fn) filter_f) (TupleInt___init__ (Int___init__ 0) idx_fn) :ruleset array_api_ruleset) +(rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ k) idx_fn) filter_f) (TupleInt_if_ (unstable-app filter_f (unstable-app idx_fn (Int___init__ k))) (TupleInt___add__ (TupleInt_single (unstable-app idx_fn (Int___init__ k))) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) :when ((!= k 0)) :ruleset array_api_ruleset) +(function bottom_indexing (Int) Int) +(rewrite (TupleInt_EMPTY) (TupleInt___init__ (Int___init__ 0) (unstable-fn "bottom_indexing")) :ruleset array_api_ruleset) +(rewrite (TupleInt_if_ (TRUE) ti ti2) ti :ruleset array_api_ruleset) +(rewrite (TupleInt_if_ (FALSE) ti ti2) ti2 :ruleset array_api_ruleset) +(rewrite (Float_abs (Float___init__ f)) (Float___init__ f) :when ((>= f 0.0)) :ruleset array_api_ruleset) +(rewrite (Float_abs (Float___init__ f)) (Float___init__ (neg f)) :when ((< f 0.0)) :ruleset array_api_ruleset) +(function Float_rational (Rational) Float :cost 2) +(rewrite (Float___init__ f) (Float_rational (rational (to-i64 f) 1)) :when ((= (to-f64 (to-i64 f)) f)) :ruleset array_api_ruleset) +(function Float_from_int (Int) Float) +(rewrite (Float_from_int (Int___init__ i)) (Float_rational (rational i 1)) :ruleset array_api_ruleset) +(function Float___add__ (Float Float) Float) +(rewrite (Float___add__ (Float___init__ f) (Float___init__ f2)) (Float___init__ (+ f f2)) :ruleset array_api_ruleset) +(function Float___sub__ (Float Float) Float) +(rewrite (Float___sub__ (Float___init__ f) (Float___init__ f2)) (Float___init__ (- f f2)) :ruleset array_api_ruleset) +(function Float___mul__ (Float Float) Float) +(rewrite (Float___mul__ (Float___init__ f) (Float___init__ f2)) (Float___init__ (* f f2)) :ruleset array_api_ruleset) +(function Float___truediv__ (Float Float) Float) +(rewrite (Float___truediv__ (Float_rational r) (Float_rational r1)) (Float_rational (/ r r1)) :ruleset array_api_ruleset) +(rewrite (Float___add__ (Float_rational r) (Float_rational r1)) (Float_rational (+ r r1)) :ruleset array_api_ruleset) +(rewrite (Float___sub__ (Float_rational r) (Float_rational r1)) (Float_rational (- r r1)) :ruleset array_api_ruleset) +(rewrite (Float___mul__ (Float_rational r) (Float_rational r1)) (Float_rational (* r r1)) :ruleset array_api_ruleset) +(rewrite (Int___eq__ (Int___init__ i) (Int___init__ i)) (TRUE) :ruleset array_api_ruleset) +(rule ((= r (Int___eq__ (Int___init__ i) (Int___init__ j))) + (!= i j)) + ((union r (FALSE))) + :ruleset array_api_ruleset ) +(function Int___ge__ (Int Int) Boolean) +(rewrite (Int___ge__ (Int___init__ i) (Int___init__ i)) (TRUE) :ruleset array_api_ruleset) +(rule ((= r (Int___ge__ (Int___init__ i) (Int___init__ j))) + (> i j)) + ((union r (TRUE))) + :ruleset array_api_ruleset ) +(rule ((= r (Int___ge__ (Int___init__ i) (Int___init__ j))) + (< i j)) + ((union r (FALSE))) + :ruleset array_api_ruleset ) +(rewrite (Int___lt__ (Int___init__ i) (Int___init__ i)) (FALSE) :ruleset array_api_ruleset) +(rule ((= r (Int___lt__ (Int___init__ i) (Int___init__ j))) + (< i j)) + ((union r (TRUE))) + :ruleset array_api_ruleset ) +(rule ((= r (Int___lt__ (Int___init__ i) (Int___init__ j))) + (> i j)) + ((union r (FALSE))) + :ruleset array_api_ruleset ) +(function Int___gt__ (Int Int) Boolean) +(rewrite (Int___gt__ (Int___init__ i) (Int___init__ i)) (FALSE) :ruleset array_api_ruleset) +(rule ((= r (Int___gt__ (Int___init__ i) (Int___init__ j))) + (> i j)) + ((union r (TRUE))) + :ruleset array_api_ruleset ) +(rule ((= r (Int___gt__ (Int___init__ i) (Int___init__ j))) + (< i j)) + ((union r (FALSE))) + :ruleset array_api_ruleset ) +(function Int_i64 (Int) i64) +(rule ((= o (Int___init__ j))) + ((set (Int_i64 o) j)) + :ruleset array_api_ruleset ) +(rewrite (Int___add__ (Int___init__ i) (Int___init__ j)) (Int___init__ (+ i j)) :ruleset array_api_ruleset) +(rewrite (Int___sub__ (Int___init__ i) (Int___init__ j)) (Int___init__ (- i j)) :ruleset array_api_ruleset) +(rewrite (Int___mul__ (Int___init__ i) (Int___init__ j)) (Int___init__ (* i j)) :ruleset array_api_ruleset) +(function Int___truediv__ (Int Int) Int) +(rewrite (Int___truediv__ (Int___init__ i) (Int___init__ j)) (Int___init__ (/ i j)) :ruleset array_api_ruleset) +(function Int___mod__ (Int Int) Int) +(rewrite (Int___mod__ (Int___init__ i) (Int___init__ j)) (Int___init__ (% i j)) :ruleset array_api_ruleset) +(function Int___and__ (Int Int) Int) +(rewrite (Int___and__ (Int___init__ i) (Int___init__ j)) (Int___init__ (& i j)) :ruleset array_api_ruleset) +(function Int___or__ (Int Int) Int) +(rewrite (Int___or__ (Int___init__ i) (Int___init__ j)) (Int___init__ (| i j)) :ruleset array_api_ruleset) +(function Int___xor__ (Int Int) Int) +(rewrite (Int___xor__ (Int___init__ i) (Int___init__ j)) (Int___init__ (^ i j)) :ruleset array_api_ruleset) +(function Int___lshift__ (Int Int) Int) +(rewrite (Int___lshift__ (Int___init__ i) (Int___init__ j)) (Int___init__ (<< i j)) :ruleset array_api_ruleset) +(function Int___rshift__ (Int Int) Int) +(rewrite (Int___rshift__ (Int___init__ i) (Int___init__ j)) (Int___init__ (>> i j)) :ruleset array_api_ruleset) +(function Int___invert__ (Int) Int) +(rewrite (Int___invert__ (Int___init__ i)) (Int___init__ (not-i64 i)) :ruleset array_api_ruleset) +(rewrite (Int_if_ (TRUE) o b) o :ruleset array_api_ruleset) +(rewrite (Int_if_ (FALSE) o b) b :ruleset array_api_ruleset) +(function Boolean_bool (Boolean) bool) +(rule ((= x (TRUE))) + ((set (Boolean_bool x) true)) + :ruleset array_api_ruleset ) +(rule ((= x (FALSE))) + ((set (Boolean_bool x) false)) + :ruleset array_api_ruleset ) +(rewrite (Boolean___or__ (TRUE) x) (TRUE) :ruleset array_api_ruleset) +(rewrite (Boolean___or__ (FALSE) x) x :ruleset array_api_ruleset) +(function Boolean___and__ (Boolean Boolean) Boolean) +(rewrite (Boolean___and__ (TRUE) x) x :ruleset array_api_ruleset) +(rewrite (Boolean___and__ (FALSE) x) (FALSE) :ruleset array_api_ruleset) +(function Boolean_if_int (Boolean Int Int) Int) +(rewrite (Boolean_if_int (TRUE) i j) i :ruleset array_api_ruleset) +(rewrite (Boolean_if_int (FALSE) i j) j :ruleset array_api_ruleset) +(function Boolean___invert__ (Boolean) Boolean) +(rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) +(rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_-4303590302573073911 (OptionalBool_none)) +(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_8855076930384822898 (OptionalDevice_none)) +(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none)) (OptionalIntOrTuple_none))))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898)) (Int___init__ 3)) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_755392963466248965 (Int___init__ 2)) +(Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_755392963466248965) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_-990525301813695487 (OptionalDType_none)) +(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_-4671014753414284575 (Int___init__ 0)) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_755392963466248965) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_-3636589015846229112 (Int___init__ 1)) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) +(run-schedule (saturate (run array_api_ruleset))) +(DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (DType_object)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_755392963466248965) +(run-schedule (saturate (run array_api_ruleset))) +(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (IsDtypeKind_string "real floating")) +(run-schedule (saturate (run array_api_ruleset))) +(Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) %__expr_-4671014753414284575) %__expr_755392963466248965) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_8800461850802719337 (Int___init__ 150)) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)) +(run-schedule (saturate (run array_api_ruleset))) +(TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_-2515197891056208817 (Value_int (Int___init__ 0))) +(Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_-2515197891056208817))))) +(run-schedule (saturate (run array_api_ruleset))) +(Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112) %__expr_755392963466248965) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575)) %__expr_-4671014753414284575) +(run-schedule (saturate (run array_api_ruleset))) +(NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_-3193103899592592052 (Value_int (Int___init__ 1))) +(let %__expr_2090748637113203634 (Value_int (Int___init__ 2))) +(let %__expr_8968271287986727247 (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(let %__expr_-346696523742155791 (TupleNDArray___getitem__ %__expr_8968271287986727247 %__expr_-3636589015846229112)) +(let %__expr_888968520189910301 (NDArray_scalar %__expr_-2515197891056208817)) +(let %__expr_8768355756772896527 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(let %__expr_-294044290527551890 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) +(let %__expr_-8993155933335339035 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) +(sort MultiAxisIndexKeyItem) +(sort Slice) +(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) +(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) +(let %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) +(let %__expr_-2454709417071966146 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) +(let %__expr_-277838035368029254 (OptionalInt_none)) +(let %__expr_-3045893352967658126 (FALSE)) +(sort MultiAxisIndexKey) +(function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) +(sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) +(function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) +(function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) +(let %__expr_469855974343835084 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-4671014753414284575) %__expr_1503884865394975690)))) +(let %__expr_5553578895160412584 (Float___init__ 1.0)) +(sort IntOrTuple) +(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) +(function IntOrTuple_int (Int) IntOrTuple) +(let %__expr_-4976863396675597554 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-4671014753414284575))) +(function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) +(function zeros (TupleInt OptionalDType OptionalDevice) NDArray) +(function OptionalDType_some (DType) OptionalDType) +(sort Device) +(function OptionalDevice_some (Device) OptionalDevice) +(function NDArray_device (NDArray) Device) +(function mean (NDArray OptionalIntOrTuple Boolean) NDArray) +(function IndexKey_ndarray (NDArray) IndexKey) +(function NDArray___eq__ (NDArray NDArray) NDArray) +(let %__expr_6782829203688770802 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device %__expr_-2454709417071966146))) %__expr_469855974343835084 (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 %__expr_888968520189910301))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_-3193103899592592052)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_2090748637113203634)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126))) +(let %__expr_-8812288226011607914 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690)))) +(let %__expr_7373748767722314024 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690)))) +(function std (NDArray OptionalIntOrTuple) NDArray) +(function OptionalInt_some (Int) OptionalInt) +(let %__expr_-4845818903945609769 (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-4671014753414284575))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575)) %__expr_-4976863396675597554)) +(let %__expr_-3351178202206693037 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-4671014753414284575))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575))) +(function NDArray___mul__ (NDArray NDArray) NDArray) +(function ndarray-sqrt (NDArray) NDArray) +(TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112) +(run-schedule (saturate (run array_api_ruleset))) +(let %__expr_6487687809380785968 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)) +(let %__expr_-3021532387541159817 (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_-277838035368029254)) +(let %__expr_5701152810033784528 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(let %__expr_-6187783262847860806 (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(let %__expr_-315001527591743086 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) +(let %__expr_-9037266425291862563 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126)) +(function NDArray___matmul__ (NDArray NDArray) NDArray) +(function IndexKey_slice (Slice) IndexKey) +(TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_5701152810033784528)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112))))))) (NDArray_T (NDArray___sub__ %__expr_6782829203688770802 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) %__expr_-3045893352967658126)) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none))))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (Int___mul__ %__expr_-3636589015846229112 %__expr_6487687809380785968) %__expr_755392963466248965) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) +(run-schedule (saturate (run array_api_ruleset))) +(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(run-schedule (saturate (run array_api_ruleset))) +(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none)) (OptionalIntOrTuple_none))))) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___lt__ %__expr_5701152810033784528 %__expr_-3636589015846229112) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) +(run-schedule (saturate (run array_api_ruleset))) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1)) +(run-schedule (saturate (run array_api_ruleset))) +(TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) +(run-schedule (saturate (run array_api_ruleset))) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_-3636589015846229112) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1))) +(run-schedule (saturate (run array_api_ruleset))) +(pop 1) (ruleset array_api_ruleset) (sort Int) -(function cast_Callable__Int__Int___Int___lambda_i_____i_ (Int Int) Int) -(rewrite (cast_Callable__Int__Int___Int___lambda_i_____i_ __var__i __var___) __var__i :ruleset array_api_ruleset) +(function i-Int__-Int_i (Int Int) Int) +(rewrite (i-Int__-Int_i i _) i :ruleset array_api_ruleset) (sort TupleInt) (function TupleInt_single (Int) TupleInt) (sort UnstableFn_Int_Int (UnstableFn (Int) Int)) -(function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt) +(function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt :unextractable) (function Int___init__ (i64) Int) -(rewrite (TupleInt_single __var__i) (TupleInt___init__ (Int___init__ 1) (unstable-fn "cast_Callable__Int__Int___Int___lambda_i_____i_" __var__i)) :ruleset array_api_ruleset) -(function cast_Callable__Int___Int___lambda_i__i_ (Int) Int) -(rewrite (cast_Callable__Int___Int___lambda_i__i_ __var__i) __var__i :ruleset array_api_ruleset) +(rewrite (TupleInt_single i) (TupleInt___init__ (Int___init__ 1) (unstable-fn "i-Int__-Int_i" i)) :ruleset array_api_ruleset) +(function i-Int_i (Int) Int) +(rewrite (i-Int_i i) i :ruleset array_api_ruleset) (function TupleInt_range (Int) TupleInt) -(rewrite (TupleInt_range __var__stop) (TupleInt___init__ __var__stop (unstable-fn "cast_Callable__Int___Int___lambda_i__i_")) :ruleset array_api_ruleset) +(rewrite (TupleInt_range stop) (TupleInt___init__ stop (unstable-fn "i-Int_i")) :ruleset array_api_ruleset) (sort Vec_Int (Vec Int)) (function TupleInt_from_vec (Vec_Int) TupleInt) (function index_vec_int (Vec_Int Int) Int) -(rewrite (TupleInt_from_vec __var__vec) (TupleInt___init__ (Int___init__ (vec-length __var__vec)) (unstable-fn "index_vec_int" __var__vec)) :ruleset array_api_ruleset) -(function cast_Callable__TupleInt__TupleInt__Int___Int___lambda_other__self__i__Int_if__i_<_self_length____self_i___other_i_-_self_length_____ (TupleInt TupleInt Int) Int) +(rewrite (TupleInt_from_vec vec) (TupleInt___init__ (Int___init__ (vec-length vec)) (unstable-fn "index_vec_int" vec)) :ruleset array_api_ruleset) (sort Boolean) (function Int_if_ (Boolean Int Int) Int) (function Int___lt__ (Int Int) Boolean) (function TupleInt_length (TupleInt) Int) (function TupleInt___getitem__ (TupleInt Int) Int) (function Int___sub__ (Int Int) Int) -(rewrite (cast_Callable__TupleInt__TupleInt__Int___Int___lambda_other__self__i__Int_if__i_<_self_length____self_i___other_i_-_self_length_____ __var__other __var__self __var__i) (Int_if_ (Int___lt__ __var__i (TupleInt_length __var__self)) (TupleInt___getitem__ __var__self __var__i) (TupleInt___getitem__ __var__other (Int___sub__ __var__i (TupleInt_length __var__self)))) :ruleset array_api_ruleset) +(function other-TupleInt_self-TupleInt_i-Int__Int_if___Int___lt___i__TupleInt_length_self____TupleInt___getitem___self_i___TupleInt___getitem___other__Int___sub___i__TupleInt_length_self____ (TupleInt TupleInt Int) Int) +(rewrite (other-TupleInt_self-TupleInt_i-Int__Int_if___Int___lt___i__TupleInt_length_self____TupleInt___getitem___self_i___TupleInt___getitem___other__Int___sub___i__TupleInt_length_self____ other self i) (Int_if_ (Int___lt__ i (TupleInt_length self)) (TupleInt___getitem__ self i) (TupleInt___getitem__ other (Int___sub__ i (TupleInt_length self)))) :ruleset array_api_ruleset) (function TupleInt___add__ (TupleInt TupleInt) TupleInt) (function Int___add__ (Int Int) Int) -(rewrite (TupleInt___add__ __var__self __var__other) (TupleInt___init__ (Int___add__ (TupleInt_length __var__self) (TupleInt_length __var__other)) (unstable-fn "cast_Callable__TupleInt__TupleInt__Int___Int___lambda_other__self__i__Int_if__i_<_self_length____self_i___other_i_-_self_length_____" __var__other __var__self)) :ruleset array_api_ruleset) -(function cast_Callable__Int__Boolean__Int___Boolean___lambda_i__acc__j__acc_|__i_==_j__ (Int Boolean Int) Boolean) +(rewrite (TupleInt___add__ self other) (TupleInt___init__ (Int___add__ (TupleInt_length self) (TupleInt_length other)) (unstable-fn "other-TupleInt_self-TupleInt_i-Int__Int_if___Int___lt___i__TupleInt_length_self____TupleInt___getitem___self_i___TupleInt___getitem___other__Int___sub___i__TupleInt_length_self____" other self)) :ruleset array_api_ruleset) (function Boolean___or__ (Boolean Boolean) Boolean) (function Int___eq__ (Int Int) Boolean) -(rewrite (cast_Callable__Int__Boolean__Int___Boolean___lambda_i__acc__j__acc_|__i_==_j__ __var__i __var__acc __var__j) (Boolean___or__ __var__acc (Int___eq__ __var__i __var__j)) :ruleset array_api_ruleset) +(function i-Int_acc-Boolean_j-Int__Boolean___or___acc__Int___eq___i_j__ (Int Boolean Int) Boolean) +(rewrite (i-Int_acc-Boolean_j-Int__Boolean___or___acc__Int___eq___i_j__ i acc j) (Boolean___or__ acc (Int___eq__ i j)) :ruleset array_api_ruleset) (function TupleInt_contains (TupleInt Int) Boolean) (sort UnstableFn_Boolean_Boolean_Int (UnstableFn (Boolean Int) Boolean)) (function TupleInt_fold_boolean (TupleInt Boolean UnstableFn_Boolean_Boolean_Int) Boolean) (function FALSE () Boolean) -(rewrite (TupleInt_contains __var__self __var__i) (TupleInt_fold_boolean __var__self (FALSE) (unstable-fn "cast_Callable__Int__Boolean__Int___Boolean___lambda_i__acc__j__acc_|__i_==_j__" __var__i)) :ruleset array_api_ruleset) +(rewrite (TupleInt_contains self i) (TupleInt_fold_boolean self (FALSE) (unstable-fn "i-Int_acc-Boolean_j-Int__Boolean___or___acc__Int___eq___i_j__" i)) :ruleset array_api_ruleset) +(function f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__ (UnstableFn_Int_Int TupleInt Int) Int) +(rewrite (f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__ f self i) (unstable-app f (TupleInt___getitem__ self i)) :ruleset array_api_ruleset) +(function TupleInt_map (TupleInt UnstableFn_Int_Int) TupleInt :cost 100) +(rewrite (TupleInt_map self f) (TupleInt___init__ (TupleInt_length self) (unstable-fn "f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__" f self)) :ruleset array_api_ruleset) +(function i-TupleInt__-Int_i (TupleInt Int) TupleInt) +(rewrite (i-TupleInt__-Int_i i _) i :ruleset array_api_ruleset) +(sort TupleTupleInt) +(function TupleTupleInt_single (TupleInt) TupleTupleInt) +(sort UnstableFn_TupleInt_Int (UnstableFn (Int) TupleInt)) +(function TupleTupleInt___init__ (Int UnstableFn_TupleInt_Int) TupleTupleInt) +(rewrite (TupleTupleInt_single i) (TupleTupleInt___init__ (Int___init__ 1) (unstable-fn "i-TupleInt__-Int_i" i)) :ruleset array_api_ruleset) +(function TupleTupleInt_from_vec (Vec_Int) TupleInt) +(rewrite (TupleTupleInt_from_vec vec) (TupleInt___init__ (Int___init__ (vec-length vec)) (unstable-fn "index_vec_int" vec)) :ruleset array_api_ruleset) +(function TupleInt_if_ (Boolean TupleInt TupleInt) TupleInt) +(function TupleTupleInt_length (TupleTupleInt) Int) +(function TupleTupleInt___getitem__ (TupleTupleInt Int) TupleInt) +(function other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ (TupleTupleInt TupleTupleInt Int) TupleInt) +(rewrite (other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ other self i) (TupleInt_if_ (Int___lt__ i (TupleTupleInt_length self)) (TupleTupleInt___getitem__ self i) (TupleTupleInt___getitem__ other (Int___sub__ i (TupleTupleInt_length self)))) :ruleset array_api_ruleset) +(function TupleTupleInt___add__ (TupleTupleInt TupleTupleInt) TupleTupleInt) +(rewrite (TupleTupleInt___add__ self other) (TupleTupleInt___init__ (Int___add__ (TupleTupleInt_length self) (TupleTupleInt_length other)) (unstable-fn "other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____" other self)) :ruleset array_api_ruleset) (sort NDArray) (function NDArray_size (NDArray) Int) (sort UnstableFn_Int_Int_Int (UnstableFn (Int Int) Int)) (function TupleInt_fold (TupleInt Int UnstableFn_Int_Int_Int) Int) (function NDArray_shape (NDArray) TupleInt) (function Int___mul__ (Int Int) Int) -(rewrite (NDArray_size __var__x) (TupleInt_fold (NDArray_shape __var__x) (Int___init__ 1) (unstable-fn "Int___mul__")) :ruleset array_api_ruleset) +(rewrite (NDArray_size x) (TupleInt_fold (NDArray_shape x) (Int___init__ 1) (unstable-fn "Int___mul__")) :ruleset array_api_ruleset) (function unique_values (NDArray) NDArray) (sort TupleValue) (function NDArray_vector (TupleValue) NDArray) @@ -50,193 +678,197 @@ (function possible_values (Value) TupleValue) (function NDArray_index (NDArray TupleInt) Value) (function ALL_INDICES () TupleInt) -(rewrite (unique_values __var__a) (NDArray_vector (possible_values (NDArray_index __var__a (ALL_INDICES)))) :ruleset array_api_ruleset) +(rewrite (unique_values a) (NDArray_vector (possible_values (NDArray_index a (ALL_INDICES)))) :ruleset array_api_ruleset) (function Value_isfinite (Value) Boolean) (function Value_int (Int) Value) (function TRUE () Boolean) -(rewrite (Value_isfinite (Value_int __var__i)) (TRUE) :ruleset array_api_ruleset) +(rewrite (Value_isfinite (Value_int i)) (TRUE) :ruleset array_api_ruleset) (function Value_bool (Boolean) Value) -(rewrite (Value_isfinite (Value_bool __var__b)) (TRUE) :ruleset array_api_ruleset) +(rewrite (Value_isfinite (Value_bool b)) (TRUE) :ruleset array_api_ruleset) (sort Float) (function Value_float (Float) Value) (function Float___init__ (f64) Float :cost 3) -(rewrite (Value_isfinite (Value_float (Float___init__ __var__f))) (TRUE) :when ((!= __var__f NaN)) :ruleset array_api_ruleset) +(rewrite (Value_isfinite (Value_float (Float___init__ f))) (TRUE) :when ((!= f NaN)) :ruleset array_api_ruleset) (function isfinite (NDArray) NDArray) (sort OptionalIntOrTuple) (function sum (NDArray OptionalIntOrTuple) NDArray) (function OptionalIntOrTuple_none () OptionalIntOrTuple) (function NDArray_scalar (Value) NDArray) -(rewrite (isfinite (sum __var__arr (OptionalIntOrTuple_none))) (NDArray_scalar (Value_bool (Value_isfinite (NDArray_index __var__arr (ALL_INDICES))))) :ruleset array_api_ruleset) +(rewrite (isfinite (sum arr (OptionalIntOrTuple_none))) (NDArray_scalar (Value_bool (Value_isfinite (NDArray_index arr (ALL_INDICES))))) :ruleset array_api_ruleset) (function assume_value_one_of (NDArray TupleValue) NDArray) -(rewrite (NDArray_shape (assume_value_one_of __var__x __var__vs)) (NDArray_shape __var__x) :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_value_one_of x vs)) (NDArray_shape x) :ruleset array_api_ruleset) (sort DType) (function NDArray_dtype (NDArray) DType) -(rewrite (NDArray_dtype (assume_value_one_of __var__x __var__vs)) (NDArray_dtype __var__x) :ruleset array_api_ruleset) -(rule ((= __var__v (NDArray_index (assume_value_one_of __var__x __var__vs) __var__idx))) - ((union __var__v (NDArray_index __var__x __var__idx)) - (union (possible_values __var__v) __var__vs)) +(rewrite (NDArray_dtype (assume_value_one_of x vs)) (NDArray_dtype x) :ruleset array_api_ruleset) +(rule ((= v (NDArray_index (assume_value_one_of x vs) idx))) + ((union v (NDArray_index x idx)) + (union (possible_values v) vs)) :ruleset array_api_ruleset ) (function assume_isfinite (NDArray) NDArray) -(rewrite (NDArray_shape (assume_isfinite __var__x)) (NDArray_shape __var__x) :ruleset array_api_ruleset) -(rewrite (NDArray_dtype (assume_isfinite __var__x)) (NDArray_dtype __var__x) :ruleset array_api_ruleset) -(rewrite (NDArray_index (assume_isfinite __var__x) __var__ti) (NDArray_index __var__x __var__ti) :ruleset array_api_ruleset) -(rewrite (Value_isfinite (NDArray_index (assume_isfinite __var__x) __var__ti)) (TRUE) :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_isfinite x)) (NDArray_shape x) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_isfinite x)) (NDArray_dtype x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_isfinite x) ti) (NDArray_index x ti) :ruleset array_api_ruleset) +(rewrite (Value_isfinite (NDArray_index (assume_isfinite x) ti)) (TRUE) :ruleset array_api_ruleset) (function assume_shape (NDArray TupleInt) NDArray) -(rewrite (NDArray_shape (assume_shape __var__x __var__shape)) __var__shape :ruleset array_api_ruleset) -(rewrite (NDArray_dtype (assume_shape __var__x __var__shape)) (NDArray_dtype __var__x) :ruleset array_api_ruleset) -(rewrite (NDArray_index (assume_shape __var__x __var__shape) __var__idx) (NDArray_index __var__x __var__idx) :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_shape x shape)) shape :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_shape x shape)) (NDArray_dtype x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_shape x shape) idx) (NDArray_index x idx) :ruleset array_api_ruleset) (function assume_dtype (NDArray DType) NDArray) -(rewrite (NDArray_dtype (assume_dtype __var__x __var__dtype)) __var__dtype :ruleset array_api_ruleset) -(rewrite (NDArray_shape (assume_dtype __var__x __var__dtype)) (NDArray_shape __var__x) :ruleset array_api_ruleset) -(rewrite (NDArray_index (assume_dtype __var__x __var__dtype) __var__idx) (NDArray_index __var__x __var__idx) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_dtype x dtype)) dtype :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_dtype x dtype)) (NDArray_shape x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_dtype x dtype) idx) (NDArray_index x idx) :ruleset array_api_ruleset) (sort IndexKey) (function NDArray___getitem__ (NDArray IndexKey) NDArray) (function IndexKey_int (Int) IndexKey) -(rewrite (NDArray___getitem__ __var__x (IndexKey_int __var__i)) (NDArray_scalar (NDArray_index __var__x (TupleInt_single __var__i))) :ruleset array_api_ruleset) +(rewrite (NDArray___getitem__ x (IndexKey_int i)) (NDArray_scalar (NDArray_index x (TupleInt_single i))) :ruleset array_api_ruleset) (sort OptionalBool) (function reshape (NDArray TupleInt OptionalBool) NDArray) -(rule ((= __a (reshape __var__x __var__shape __var__copy))) - ((NDArray_shape __var__x) - (TupleInt_length (NDArray_shape __var__x))) +(rule ((= __a (reshape x shape copy))) + ((NDArray_shape x) + (TupleInt_length (NDArray_shape x))) :ruleset array_api_ruleset ) -(rule ((reshape __var__x __var__shape __var__copy)) - ((TupleInt_length __var__shape) - (TupleInt___getitem__ __var__shape (Int___init__ 0))) +(rule ((reshape x shape copy)) + ((TupleInt_length shape) + (TupleInt___getitem__ shape (Int___init__ 0))) :ruleset array_api_ruleset ) -(rewrite (reshape __var__x __var__shape __var__copy) __var__x :when ((= (TupleInt_length (NDArray_shape __var__x)) (Int___init__ 1)) (= (TupleInt_length __var__shape) (Int___init__ 1)) (= (TupleInt___getitem__ __var__shape (Int___init__ 0)) (Int___init__ -1))) :ruleset array_api_ruleset) +(rewrite (reshape x shape copy) x :when ((= (TupleInt_length (NDArray_shape x)) (Int___init__ 1)) (= (TupleInt_length shape) (Int___init__ 1)) (= (TupleInt___getitem__ shape (Int___init__ 0)) (Int___init__ -1))) :ruleset array_api_ruleset) (function TupleValue_length (TupleValue) Int) -(rewrite (NDArray_shape (NDArray_vector __var__vs)) (TupleInt_single (TupleValue_length __var__vs)) :ruleset array_api_ruleset) +(rewrite (NDArray_shape (NDArray_vector vs)) (TupleInt_single (TupleValue_length vs)) :ruleset array_api_ruleset) (function Value_dtype (Value) DType) (function TupleValue___getitem__ (TupleValue Int) Value) -(rewrite (NDArray_dtype (NDArray_vector __var__vs)) (Value_dtype (TupleValue___getitem__ __var__vs (Int___init__ 0))) :ruleset array_api_ruleset) -(rewrite (NDArray_index (NDArray_vector __var__vs) __var__ti) (TupleValue___getitem__ __var__vs (TupleInt___getitem__ __var__ti (Int___init__ 0))) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (NDArray_vector vs)) (Value_dtype (TupleValue___getitem__ vs (Int___init__ 0))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_vector vs) ti) (TupleValue___getitem__ vs (TupleInt___getitem__ ti (Int___init__ 0))) :ruleset array_api_ruleset) (function TupleInt_EMPTY () TupleInt) -(rewrite (NDArray_shape (NDArray_scalar __var__v)) (TupleInt_EMPTY) :ruleset array_api_ruleset) -(rewrite (NDArray_dtype (NDArray_scalar __var__v)) (Value_dtype __var__v) :ruleset array_api_ruleset) -(rewrite (NDArray_index (NDArray_scalar __var__v) (TupleInt_EMPTY)) __var__v :ruleset array_api_ruleset) +(rewrite (NDArray_shape (NDArray_scalar v)) (TupleInt_EMPTY) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (NDArray_scalar v)) (Value_dtype v) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_scalar v) (TupleInt_EMPTY)) v :ruleset array_api_ruleset) (function any (NDArray) NDArray) (function TupleValue_includes (TupleValue Value) Boolean) (function Value_to_truthy_value (Value) Value) -(rewrite (any __var__x) (NDArray_scalar (Value_bool (TupleValue_includes (possible_values (Value_to_truthy_value (NDArray_index __var__x (ALL_INDICES)))) (Value_bool (TRUE))))) :ruleset array_api_ruleset) +(rewrite (any x) (NDArray_scalar (Value_bool (TupleValue_includes (possible_values (Value_to_truthy_value (NDArray_index x (ALL_INDICES)))) (Value_bool (TRUE))))) :ruleset array_api_ruleset) (function NDArray___lt__ (NDArray NDArray) NDArray) (function Value___lt__ (Value Value) Value) (function broadcast_index (TupleInt TupleInt TupleInt) TupleInt) (function broadcast_shapes (TupleInt TupleInt) TupleInt) -(rewrite (NDArray_index (NDArray___lt__ __var__x __var__y) __var__idx) (Value___lt__ (NDArray_index __var__x (broadcast_index (NDArray_shape __var__x) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx)) (NDArray_index __var__y (broadcast_index (NDArray_shape __var__y) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray___lt__ x y) idx) (Value___lt__ (NDArray_index x (broadcast_index (NDArray_shape x) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx)) (NDArray_index y (broadcast_index (NDArray_shape y) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx))) :ruleset array_api_ruleset) (function NDArray___truediv__ (NDArray NDArray) NDArray) (function Value___truediv__ (Value Value) Value) -(rewrite (NDArray_index (NDArray___truediv__ __var__x __var__y) __var__idx) (Value___truediv__ (NDArray_index __var__x (broadcast_index (NDArray_shape __var__x) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx)) (NDArray_index __var__y (broadcast_index (NDArray_shape __var__y) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx))) :ruleset array_api_ruleset) -(rewrite (NDArray_index (NDArray_scalar __var__v) __var__idx) __var__v :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray___truediv__ x y) idx) (Value___truediv__ (NDArray_index x (broadcast_index (NDArray_shape x) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx)) (NDArray_index y (broadcast_index (NDArray_shape y) (broadcast_shapes (NDArray_shape x) (NDArray_shape y)) idx))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_scalar v) idx) v :ruleset array_api_ruleset) (function astype (NDArray DType) NDArray) (function Value_astype (Value DType) Value) -(rewrite (NDArray_index (astype __var__x __var__dtype) __var__idx) (Value_astype (NDArray_index __var__x __var__idx) __var__dtype) :ruleset array_api_ruleset) +(rewrite (NDArray_index (astype x dtype) idx) (Value_astype (NDArray_index x idx) dtype) :ruleset array_api_ruleset) (relation greater_zero (Value)) (sort TupleNDArray) (function TupleNDArray___getitem__ (TupleNDArray Int) NDArray) (function unique_counts (NDArray) TupleNDArray) -(rule ((= __var__v (NDArray_index (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) __var__idx))) - ((greater_zero __var__v)) +(rule ((= v (NDArray_index (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) idx))) + ((greater_zero v)) :ruleset array_api_ruleset ) -(rule ((greater_zero __var__v) - (= __var__v1 (Value_astype __var__v __var__dtype))) - ((greater_zero __var__v1)) +(rule ((greater_zero v) + (= v1 (Value_astype v dtype))) + ((greater_zero v1)) :ruleset array_api_ruleset ) -(rule ((= __var__v (Value_float (Float___init__ __var__f))) - (> __var__f 0.0)) - ((greater_zero __var__v)) +(rule ((= v (Value_float (Float___init__ f))) + (> f 0.0)) + ((greater_zero v)) :ruleset array_api_ruleset ) -(rule ((= __var__v (Value_int (Int___init__ __var__i))) - (> __var__i 0)) - ((greater_zero __var__v)) +(rule ((= v (Value_int (Int___init__ i))) + (> i 0)) + ((greater_zero v)) :ruleset array_api_ruleset ) -(rule ((greater_zero __var__v) - (greater_zero __var__v1) - (= __var__v2 (Value___truediv__ __var__v __var__v1))) - ((greater_zero __var__v2)) +(rule ((greater_zero v) + (greater_zero v1) + (= v2 (Value___truediv__ v v1))) + ((greater_zero v2)) :ruleset array_api_ruleset ) -(rule ((greater_zero __var__v) - (= __var__v1 (Value___lt__ __var__v (Value_int (Int___init__ 0))))) - ((union __var__v1 (Value_bool (FALSE)))) +(rule ((greater_zero v) + (= v1 (Value___lt__ v (Value_int (Int___init__ 0))))) + ((union v1 (Value_bool (FALSE)))) :ruleset array_api_ruleset ) (function TupleValue___init__ (Value) TupleValue) -(rewrite (possible_values (Value_bool __var__b)) (TupleValue___init__ (Value_bool __var__b)) :ruleset array_api_ruleset) -(rule ((= __var__v1 (Value_astype __var__v __var__dtype)) - (greater_zero __var__v)) - ((greater_zero __var__v1)) +(rewrite (possible_values (Value_bool b)) (TupleValue___init__ (Value_bool b)) :ruleset array_api_ruleset) +(rule ((= v1 (Value_astype v dtype)) + (greater_zero v)) + ((greater_zero v1)) :ruleset array_api_ruleset ) (function TupleNDArray_length (TupleNDArray) Int) (function svd (NDArray Boolean) TupleNDArray) -(rewrite (TupleNDArray_length (svd __var__x __var__full_matrices)) (Int___init__ 3) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (svd x full_matrices)) (Int___init__ 3) :ruleset array_api_ruleset) (function unique_inverse (NDArray) TupleNDArray) -(rewrite (TupleNDArray_length (unique_inverse __var__x)) (Int___init__ 2) :ruleset array_api_ruleset) -(rewrite (TupleNDArray___getitem__ (unique_inverse __var__x) (Int___init__ 0)) (unique_values __var__x) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (unique_inverse x)) (Int___init__ 2) :ruleset array_api_ruleset) +(rewrite (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 0)) (unique_values x) :ruleset array_api_ruleset) (function ndarray-abs (NDArray) NDArray) (function Float_abs (Float) Float) -(rewrite (ndarray-abs (NDArray_scalar (Value_float __var__f))) (NDArray_scalar (Value_float (Float_abs __var__f))) :ruleset array_api_ruleset) -(rewrite (TupleNDArray_length (unique_counts __var__x)) (Int___init__ 2) :ruleset array_api_ruleset) -(rewrite (sum (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_int (NDArray_size __var__x))) :ruleset array_api_ruleset) -(rewrite (sum (astype (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) __var__dtype) (OptionalIntOrTuple_none)) (astype (NDArray_scalar (Value_int (NDArray_size __var__x))) __var__dtype) :ruleset array_api_ruleset) -(rewrite (NDArray_dtype (astype __var__x __var__dtype)) __var__dtype :ruleset array_api_ruleset) +(rewrite (ndarray-abs (NDArray_scalar (Value_float f))) (NDArray_scalar (Value_float (Float_abs f))) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (unique_counts x)) (Int___init__ 2) :ruleset array_api_ruleset) +(rewrite (sum (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_int (NDArray_size x))) :ruleset array_api_ruleset) +(rewrite (sum (astype (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) dtype) (OptionalIntOrTuple_none)) (astype (NDArray_scalar (Value_int (NDArray_size x))) dtype) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (astype x dtype)) dtype :ruleset array_api_ruleset) (function DType_float64 () DType) -(rewrite (astype (NDArray_scalar (Value_int (Int___init__ __var__i))) (DType_float64)) (NDArray_scalar (Value_float (Float___init__ (to-f64 __var__i)))) :ruleset array_api_ruleset) +(rewrite (astype (NDArray_scalar (Value_int (Int___init__ i))) (DType_float64)) (NDArray_scalar (Value_float (Float___init__ (to-f64 i)))) :ruleset array_api_ruleset) (sort OptionalInt) (function concat (TupleNDArray OptionalInt) NDArray) (function TupleNDArray___init__ (NDArray) TupleNDArray) (function OptionalInt_none () OptionalInt) -(rewrite (concat (TupleNDArray___init__ __var__x) (OptionalInt_none)) __var__x :ruleset array_api_ruleset) -(rewrite (unique_values (unique_values __var__x)) (unique_values __var__x) :ruleset array_api_ruleset) -(rewrite (sum (NDArray___truediv__ __var__x (NDArray_scalar __var__v)) (OptionalIntOrTuple_none)) (NDArray___truediv__ (sum __var__x (OptionalIntOrTuple_none)) (NDArray_scalar __var__v)) :ruleset array_api_ruleset) +(rewrite (concat (TupleNDArray___init__ x) (OptionalInt_none)) x :ruleset array_api_ruleset) +(rewrite (unique_values (unique_values x)) (unique_values x) :ruleset array_api_ruleset) +(rewrite (sum (NDArray___truediv__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_none)) (NDArray_scalar v)) :ruleset array_api_ruleset) (function NDArray_ndim (NDArray) Int) (sort OptionalDType) (sort OptionalDevice) (function asarray (NDArray OptionalDType OptionalBool OptionalDevice) NDArray) (function OptionalDevice_none () OptionalDevice) -(rewrite (NDArray_ndim (asarray __var__a __var__d __var__ob (OptionalDevice_none))) (NDArray_ndim __var__a) :ruleset array_api_ruleset) +(rewrite (NDArray_ndim (asarray a d ob (OptionalDevice_none))) (NDArray_ndim a) :ruleset array_api_ruleset) (function OptionalDType_none () OptionalDType) (function OptionalBool_none () OptionalBool) -(rewrite (asarray __var__a (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) __var__a :ruleset array_api_ruleset) +(rewrite (asarray a (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) a :ruleset array_api_ruleset) (function TupleNDArray___add__ (TupleNDArray TupleNDArray) TupleNDArray) (function TupleNDArray_EMPTY () TupleNDArray) -(rewrite (TupleNDArray___add__ __var__ti (TupleNDArray_EMPTY)) __var__ti :ruleset array_api_ruleset) -(rewrite (TupleNDArray_length (TupleNDArray___init__ __var__n)) (Int___init__ 1) :ruleset array_api_ruleset) -(rewrite (TupleNDArray_length (TupleNDArray___add__ __var__ti __var__ti2)) (Int___add__ (TupleNDArray_length __var__ti) (TupleNDArray_length __var__ti2)) :ruleset array_api_ruleset) -(rewrite (NDArray_ndim __var__x) (TupleInt_length (NDArray_shape __var__x)) :ruleset array_api_ruleset) +(rewrite (TupleNDArray___add__ ti (TupleNDArray_EMPTY)) ti :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (TupleNDArray___init__ n)) (Int___init__ 1) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (TupleNDArray___add__ ti ti2)) (Int___add__ (TupleNDArray_length ti) (TupleNDArray_length ti2)) :ruleset array_api_ruleset) +(sort UnstableFn_Value_TupleInt (UnstableFn (TupleInt) Value)) +(function NDArray___init__ (TupleInt DType UnstableFn_Value_TupleInt) NDArray) +(rewrite (NDArray_shape (NDArray___init__ shape dtype idx_fn)) shape :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (NDArray___init__ shape dtype idx_fn)) dtype :ruleset array_api_ruleset) +(rewrite (NDArray_ndim x) (TupleInt_length (NDArray_shape x)) :ruleset array_api_ruleset) (function NDArray_to_value (NDArray) Value) -(rewrite (NDArray_to_value __var__x) (NDArray_index __var__x (TupleInt_EMPTY)) :ruleset array_api_ruleset) -(rewrite (NDArray___truediv__ (NDArray_scalar (Value_float __var__f)) (NDArray_scalar (Value_float __var__f))) (NDArray_scalar (Value_float (Float___init__ 1.0))) :ruleset array_api_ruleset) +(rewrite (NDArray_to_value x) (NDArray_index x (TupleInt_EMPTY)) :ruleset array_api_ruleset) +(rewrite (NDArray___truediv__ (NDArray_scalar (Value_float f)) (NDArray_scalar (Value_float f))) (NDArray_scalar (Value_float (Float___init__ 1.0))) :ruleset array_api_ruleset) (function NDArray___sub__ (NDArray NDArray) NDArray) -(rewrite (NDArray___sub__ (NDArray_scalar (Value_float __var__f)) (NDArray_scalar (Value_float __var__f))) (NDArray_scalar (Value_float (Float___init__ 0.0))) :ruleset array_api_ruleset) +(rewrite (NDArray___sub__ (NDArray_scalar (Value_float f)) (NDArray_scalar (Value_float f))) (NDArray_scalar (Value_float (Float___init__ 0.0))) :ruleset array_api_ruleset) (function NDArray___gt__ (NDArray NDArray) NDArray) -(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ __var__fi1))) (NDArray_scalar (Value_float (Float___init__ __var__fi2)))) (NDArray_scalar (Value_bool (TRUE))) :when ((> __var__fi1 __var__fi2)) :ruleset array_api_ruleset) -(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ __var__fi1))) (NDArray_scalar (Value_float (Float___init__ __var__fi2)))) (NDArray_scalar (Value_bool (FALSE))) :when ((<= __var__fi1 __var__fi2)) :ruleset array_api_ruleset) +(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ fi1))) (NDArray_scalar (Value_float (Float___init__ fi2)))) (NDArray_scalar (Value_bool (TRUE))) :when ((> fi1 fi2)) :ruleset array_api_ruleset) +(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ fi1))) (NDArray_scalar (Value_float (Float___init__ fi2)))) (NDArray_scalar (Value_bool (FALSE))) :when ((<= fi1 fi2)) :ruleset array_api_ruleset) (function NDArray_T (NDArray) NDArray) -(rewrite (NDArray_T (NDArray_T __var__x)) __var__x :ruleset array_api_ruleset) +(rewrite (NDArray_T (NDArray_T x)) x :ruleset array_api_ruleset) (function TupleValue___add__ (TupleValue TupleValue) TupleValue) (function TupleValue_EMPTY () TupleValue) -(rewrite (TupleValue___add__ __var__ti (TupleValue_EMPTY)) __var__ti :ruleset array_api_ruleset) +(rewrite (TupleValue___add__ ti (TupleValue_EMPTY)) ti :ruleset array_api_ruleset) (rewrite (TupleValue_length (TupleValue_EMPTY)) (Int___init__ 0) :ruleset array_api_ruleset) -(rewrite (TupleValue_length (TupleValue___init__ __var__v)) (Int___init__ 1) :ruleset array_api_ruleset) -(rewrite (TupleValue_length (TupleValue___add__ __var__ti __var__ti2)) (Int___add__ (TupleValue_length __var__ti) (TupleValue_length __var__ti2)) :ruleset array_api_ruleset) -(rewrite (TupleValue___getitem__ (TupleValue___init__ __var__v) (Int___init__ 0)) __var__v :ruleset array_api_ruleset) -(rewrite (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ __var__v) __var__ti) (Int___init__ 0)) __var__v :ruleset array_api_ruleset) -(rule ((= __var__v (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ __var__v2) __var__ti) (Int___init__ __var__k))) - (> __var__k 0)) - ((union __var__v (TupleValue___getitem__ __var__ti (Int___init__ (- __var__k 1))))) - :ruleset array_api_ruleset ) -(rewrite (TupleValue_includes (TupleValue_EMPTY) __var__v) (FALSE) :ruleset array_api_ruleset) -(rewrite (TupleValue_includes (TupleValue___init__ __var__v) __var__v) (TRUE) :ruleset array_api_ruleset) -(rewrite (TupleValue_includes (TupleValue___init__ __var__v) __var__v2) (FALSE) :when ((!= __var__v __var__v2)) :ruleset array_api_ruleset) -(rewrite (TupleValue_includes (TupleValue___add__ __var__ti __var__ti2) __var__v) (Boolean___or__ (TupleValue_includes __var__ti __var__v) (TupleValue_includes __var__ti2 __var__v)) :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue___init__ v)) (Int___init__ 1) :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue___add__ ti ti2)) (Int___add__ (TupleValue_length ti) (TupleValue_length ti2)) :ruleset array_api_ruleset) +(rewrite (TupleValue___getitem__ (TupleValue___init__ v) (Int___init__ 0)) v :ruleset array_api_ruleset) +(rewrite (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ v) ti) (Int___init__ 0)) v :ruleset array_api_ruleset) +(rule ((= v (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ v2) ti) (Int___init__ k))) + (> k 0)) + ((union v (TupleValue___getitem__ ti (Int___init__ (- k 1))))) + :ruleset array_api_ruleset ) +(rewrite (TupleValue_includes (TupleValue_EMPTY) v) (FALSE) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___init__ v) v) (TRUE) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___init__ v) v2) (FALSE) :when ((!= v v2)) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___add__ ti ti2) v) (Boolean___or__ (TupleValue_includes ti v) (TupleValue_includes ti2 v)) :ruleset array_api_ruleset) (function DType_int64 () DType) -(rewrite (Value_dtype (Value_int __var__i)) (DType_int64) :ruleset array_api_ruleset) -(rewrite (Value_dtype (Value_float __var__f)) (DType_float64) :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_int i)) (DType_int64) :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_float f)) (DType_float64) :ruleset array_api_ruleset) (function DType_bool () DType) -(rewrite (Value_dtype (Value_bool __var__b)) (DType_bool) :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_bool b)) (DType_bool) :ruleset array_api_ruleset) (function Value_to_bool (Value) Boolean) -(rewrite (Value_to_bool (Value_bool __var__b)) __var__b :ruleset array_api_ruleset) +(rewrite (Value_to_bool (Value_bool b)) b :ruleset array_api_ruleset) (function Value_to_int (Value) Int) -(rewrite (Value_to_int (Value_int __var__i)) __var__i :ruleset array_api_ruleset) -(rewrite (Value_to_truthy_value (Value_bool __var__b)) (Value_bool __var__b) :ruleset array_api_ruleset) +(rewrite (Value_to_int (Value_int i)) i :ruleset array_api_ruleset) +(rewrite (Value_to_truthy_value (Value_bool b)) (Value_bool b) :ruleset array_api_ruleset) (sort IsDtypeKind) (function isdtype (DType IsDtypeKind) Boolean) (function DType_float32 () DType) @@ -259,12 +891,12 @@ (rewrite (isdtype (DType_int64) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (function IsDtypeKind_NULL () IsDtypeKind) -(rewrite (isdtype __var__d (IsDtypeKind_NULL)) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype d (IsDtypeKind_NULL)) (FALSE) :ruleset array_api_ruleset) (function IsDtypeKind_dtype (DType) IsDtypeKind) -(rewrite (isdtype __var__d (IsDtypeKind_dtype __var__d)) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype d (IsDtypeKind_dtype d)) (TRUE) :ruleset array_api_ruleset) (function IsDtypeKind___or__ (IsDtypeKind IsDtypeKind) IsDtypeKind :cost 10) -(rewrite (isdtype __var__d (IsDtypeKind___or__ __var__k1 __var__k2)) (Boolean___or__ (isdtype __var__d __var__k1) (isdtype __var__d __var__k2)) :ruleset array_api_ruleset) -(rewrite (IsDtypeKind___or__ __var__k1 (IsDtypeKind_NULL)) __var__k1 :ruleset array_api_ruleset) +(rewrite (isdtype d (IsDtypeKind___or__ k1 k2)) (Boolean___or__ (isdtype d k1) (isdtype d k2)) :ruleset array_api_ruleset) +(rewrite (IsDtypeKind___or__ k1 (IsDtypeKind_NULL)) k1 :ruleset array_api_ruleset) (function DType___eq__ (DType DType) Boolean) (rewrite (DType___eq__ (DType_float64) (DType_float64)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float64) (DType_float32)) (FALSE) :ruleset array_api_ruleset) @@ -291,118 +923,124 @@ (rewrite (DType___eq__ (DType_object) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_int64)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_object)) (TRUE) :ruleset array_api_ruleset) -(function cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___ (UnstableFn_Int_Int Int) Int) -(rewrite (cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___ __var__idx_fn __var__i) (unstable-app __var__idx_fn (Int___add__ __var__i (Int___init__ 1))) :ruleset array_api_ruleset) -(rewrite (index_vec_int __var__vs (Int___init__ __var__k)) (vec-get __var__vs __var__k) :ruleset array_api_ruleset) -(rewrite (TupleInt_length (TupleInt___init__ __var__i __var__idx_fn)) __var__i :ruleset array_api_ruleset) -(rewrite (TupleInt___getitem__ (TupleInt___init__ __var__i __var__idx_fn) __var__i2) (unstable-app __var__idx_fn __var__i2) :ruleset array_api_ruleset) -(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ 0) __var__idx_fn) __var__i __var__f) __var__i :ruleset array_api_ruleset) -(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ __var__k) __var__idx_fn) __var__i __var__f) (unstable-app __var__f (TupleInt_fold (TupleInt___init__ (Int___init__ (- __var__k 1)) (unstable-fn "cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___" __var__idx_fn)) __var__i __var__f) (unstable-app __var__idx_fn (Int___init__ 0))) :when ((!= __var__k 0)) :ruleset array_api_ruleset) -(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ 0) __var__idx_fn) __var__b __var__bool_f) __var__b :ruleset array_api_ruleset) -(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ __var__k) __var__idx_fn) __var__b __var__bool_f) (unstable-app __var__bool_f (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ (- __var__k 1)) (unstable-fn "cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___" __var__idx_fn)) __var__b __var__bool_f) (unstable-app __var__idx_fn (Int___init__ 0))) :when ((!= __var__k 0)) :ruleset array_api_ruleset) +(function idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ (UnstableFn_Int_Int Int) Int) +(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ idx_fn i) (unstable-app idx_fn (Int___add__ i (Int___init__ 1))) :ruleset array_api_ruleset) +(rewrite (TupleInt_length (TupleInt___init__ i idx_fn)) i :ruleset array_api_ruleset) +(rewrite (TupleInt___getitem__ (TupleInt___init__ i idx_fn) i2) (unstable-app idx_fn i2) :ruleset array_api_ruleset) +(rewrite (index_vec_int vs (Int___init__ k)) (vec-get vs k) :when ((> (vec-length vs) k)) :ruleset array_api_ruleset) +(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ 0) idx_fn) i f) i :ruleset array_api_ruleset) +(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ k) idx_fn) i f) (unstable-app f (TupleInt_fold (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) i f) (unstable-app idx_fn (Int___init__ 0))) :when ((!= k 0)) :ruleset array_api_ruleset) +(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ 0) idx_fn) b bool_f) b :ruleset array_api_ruleset) +(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ k) idx_fn) b bool_f) (unstable-app bool_f (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) b bool_f) (unstable-app idx_fn (Int___init__ 0))) :when ((!= k 0)) :ruleset array_api_ruleset) +(sort UnstableFn_Boolean_Int (UnstableFn (Int) Boolean)) +(function TupleInt_filter (TupleInt UnstableFn_Boolean_Int) TupleInt :cost 100) +(rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ 0) idx_fn) filter_f) (TupleInt___init__ (Int___init__ 0) idx_fn) :ruleset array_api_ruleset) +(rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ k) idx_fn) filter_f) (TupleInt_if_ (unstable-app filter_f (unstable-app idx_fn (Int___init__ k))) (TupleInt___add__ (TupleInt_single (unstable-app idx_fn (Int___init__ k))) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) :when ((!= k 0)) :ruleset array_api_ruleset) (function bottom_indexing (Int) Int) (rewrite (TupleInt_EMPTY) (TupleInt___init__ (Int___init__ 0) (unstable-fn "bottom_indexing")) :ruleset array_api_ruleset) -(rewrite (Float_abs (Float___init__ __var__f)) (Float___init__ __var__f) :when ((>= __var__f 0.0)) :ruleset array_api_ruleset) -(rewrite (Float_abs (Float___init__ __var__f)) (Float___init__ (neg __var__f)) :when ((< __var__f 0.0)) :ruleset array_api_ruleset) +(rewrite (TupleInt_if_ (TRUE) ti ti2) ti :ruleset array_api_ruleset) +(rewrite (TupleInt_if_ (FALSE) ti ti2) ti2 :ruleset array_api_ruleset) +(rewrite (Float_abs (Float___init__ f)) (Float___init__ f) :when ((>= f 0.0)) :ruleset array_api_ruleset) +(rewrite (Float_abs (Float___init__ f)) (Float___init__ (neg f)) :when ((< f 0.0)) :ruleset array_api_ruleset) (function Float_rational (Rational) Float :cost 2) -(rewrite (Float___init__ __var__f) (Float_rational (rational (to-i64 __var__f) 1)) :when ((= (to-f64 (to-i64 __var__f)) __var__f)) :ruleset array_api_ruleset) +(rewrite (Float___init__ f) (Float_rational (rational (to-i64 f) 1)) :when ((= (to-f64 (to-i64 f)) f)) :ruleset array_api_ruleset) (function Float_from_int (Int) Float) -(rewrite (Float_from_int (Int___init__ __var__i)) (Float_rational (rational __var__i 1)) :ruleset array_api_ruleset) +(rewrite (Float_from_int (Int___init__ i)) (Float_rational (rational i 1)) :ruleset array_api_ruleset) (function Float___add__ (Float Float) Float) -(rewrite (Float___add__ (Float___init__ __var__f) (Float___init__ __var__f2)) (Float___init__ (+ __var__f __var__f2)) :ruleset array_api_ruleset) +(rewrite (Float___add__ (Float___init__ f) (Float___init__ f2)) (Float___init__ (+ f f2)) :ruleset array_api_ruleset) (function Float___sub__ (Float Float) Float) -(rewrite (Float___sub__ (Float___init__ __var__f) (Float___init__ __var__f2)) (Float___init__ (- __var__f __var__f2)) :ruleset array_api_ruleset) +(rewrite (Float___sub__ (Float___init__ f) (Float___init__ f2)) (Float___init__ (- f f2)) :ruleset array_api_ruleset) (function Float___mul__ (Float Float) Float) -(rewrite (Float___mul__ (Float___init__ __var__f) (Float___init__ __var__f2)) (Float___init__ (* __var__f __var__f2)) :ruleset array_api_ruleset) +(rewrite (Float___mul__ (Float___init__ f) (Float___init__ f2)) (Float___init__ (* f f2)) :ruleset array_api_ruleset) (function Float___truediv__ (Float Float) Float) -(rewrite (Float___truediv__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (/ __var__r __var__r1)) :ruleset array_api_ruleset) -(rewrite (Float___add__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (+ __var__r __var__r1)) :ruleset array_api_ruleset) -(rewrite (Float___sub__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (- __var__r __var__r1)) :ruleset array_api_ruleset) -(rewrite (Float___mul__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (* __var__r __var__r1)) :ruleset array_api_ruleset) -(rewrite (Int___eq__ (Int___init__ __var__i) (Int___init__ __var__i)) (TRUE) :ruleset array_api_ruleset) -(rule ((= __var__r (Int___eq__ (Int___init__ __var__i) (Int___init__ __var__j))) - (!= __var__i __var__j)) - ((union __var__r (FALSE))) +(rewrite (Float___truediv__ (Float_rational r) (Float_rational r1)) (Float_rational (/ r r1)) :ruleset array_api_ruleset) +(rewrite (Float___add__ (Float_rational r) (Float_rational r1)) (Float_rational (+ r r1)) :ruleset array_api_ruleset) +(rewrite (Float___sub__ (Float_rational r) (Float_rational r1)) (Float_rational (- r r1)) :ruleset array_api_ruleset) +(rewrite (Float___mul__ (Float_rational r) (Float_rational r1)) (Float_rational (* r r1)) :ruleset array_api_ruleset) +(rewrite (Int___eq__ (Int___init__ i) (Int___init__ i)) (TRUE) :ruleset array_api_ruleset) +(rule ((= r (Int___eq__ (Int___init__ i) (Int___init__ j))) + (!= i j)) + ((union r (FALSE))) :ruleset array_api_ruleset ) (function Int___ge__ (Int Int) Boolean) -(rewrite (Int___ge__ (Int___init__ __var__i) (Int___init__ __var__i)) (TRUE) :ruleset array_api_ruleset) -(rule ((= __var__r (Int___ge__ (Int___init__ __var__i) (Int___init__ __var__j))) - (> __var__i __var__j)) - ((union __var__r (TRUE))) +(rewrite (Int___ge__ (Int___init__ i) (Int___init__ i)) (TRUE) :ruleset array_api_ruleset) +(rule ((= r (Int___ge__ (Int___init__ i) (Int___init__ j))) + (> i j)) + ((union r (TRUE))) :ruleset array_api_ruleset ) -(rule ((= __var__r (Int___ge__ (Int___init__ __var__i) (Int___init__ __var__j))) - (< __var__i __var__j)) - ((union __var__r (FALSE))) +(rule ((= r (Int___ge__ (Int___init__ i) (Int___init__ j))) + (< i j)) + ((union r (FALSE))) :ruleset array_api_ruleset ) -(rewrite (Int___lt__ (Int___init__ __var__i) (Int___init__ __var__i)) (FALSE) :ruleset array_api_ruleset) -(rule ((= __var__r (Int___lt__ (Int___init__ __var__i) (Int___init__ __var__j))) - (< __var__i __var__j)) - ((union __var__r (TRUE))) +(rewrite (Int___lt__ (Int___init__ i) (Int___init__ i)) (FALSE) :ruleset array_api_ruleset) +(rule ((= r (Int___lt__ (Int___init__ i) (Int___init__ j))) + (< i j)) + ((union r (TRUE))) :ruleset array_api_ruleset ) -(rule ((= __var__r (Int___lt__ (Int___init__ __var__i) (Int___init__ __var__j))) - (> __var__i __var__j)) - ((union __var__r (FALSE))) +(rule ((= r (Int___lt__ (Int___init__ i) (Int___init__ j))) + (> i j)) + ((union r (FALSE))) :ruleset array_api_ruleset ) (function Int___gt__ (Int Int) Boolean) -(rewrite (Int___gt__ (Int___init__ __var__i) (Int___init__ __var__i)) (FALSE) :ruleset array_api_ruleset) -(rule ((= __var__r (Int___gt__ (Int___init__ __var__i) (Int___init__ __var__j))) - (> __var__i __var__j)) - ((union __var__r (TRUE))) +(rewrite (Int___gt__ (Int___init__ i) (Int___init__ i)) (FALSE) :ruleset array_api_ruleset) +(rule ((= r (Int___gt__ (Int___init__ i) (Int___init__ j))) + (> i j)) + ((union r (TRUE))) :ruleset array_api_ruleset ) -(rule ((= __var__r (Int___gt__ (Int___init__ __var__i) (Int___init__ __var__j))) - (< __var__i __var__j)) - ((union __var__r (FALSE))) +(rule ((= r (Int___gt__ (Int___init__ i) (Int___init__ j))) + (< i j)) + ((union r (FALSE))) :ruleset array_api_ruleset ) (function Int_i64 (Int) i64) -(rule ((= __var__o (Int___init__ __var__j))) - ((set (Int_i64 __var__o) __var__j)) +(rule ((= o (Int___init__ j))) + ((set (Int_i64 o) j)) :ruleset array_api_ruleset ) -(rewrite (Int___add__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (+ __var__i __var__j)) :ruleset array_api_ruleset) -(rewrite (Int___sub__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (- __var__i __var__j)) :ruleset array_api_ruleset) -(rewrite (Int___mul__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (* __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___add__ (Int___init__ i) (Int___init__ j)) (Int___init__ (+ i j)) :ruleset array_api_ruleset) +(rewrite (Int___sub__ (Int___init__ i) (Int___init__ j)) (Int___init__ (- i j)) :ruleset array_api_ruleset) +(rewrite (Int___mul__ (Int___init__ i) (Int___init__ j)) (Int___init__ (* i j)) :ruleset array_api_ruleset) (function Int___truediv__ (Int Int) Int) -(rewrite (Int___truediv__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (/ __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___truediv__ (Int___init__ i) (Int___init__ j)) (Int___init__ (/ i j)) :ruleset array_api_ruleset) (function Int___mod__ (Int Int) Int) -(rewrite (Int___mod__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (% __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___mod__ (Int___init__ i) (Int___init__ j)) (Int___init__ (% i j)) :ruleset array_api_ruleset) (function Int___and__ (Int Int) Int) -(rewrite (Int___and__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (& __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___and__ (Int___init__ i) (Int___init__ j)) (Int___init__ (& i j)) :ruleset array_api_ruleset) (function Int___or__ (Int Int) Int) -(rewrite (Int___or__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (| __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___or__ (Int___init__ i) (Int___init__ j)) (Int___init__ (| i j)) :ruleset array_api_ruleset) (function Int___xor__ (Int Int) Int) -(rewrite (Int___xor__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (^ __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___xor__ (Int___init__ i) (Int___init__ j)) (Int___init__ (^ i j)) :ruleset array_api_ruleset) (function Int___lshift__ (Int Int) Int) -(rewrite (Int___lshift__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (<< __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___lshift__ (Int___init__ i) (Int___init__ j)) (Int___init__ (<< i j)) :ruleset array_api_ruleset) (function Int___rshift__ (Int Int) Int) -(rewrite (Int___rshift__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (>> __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___rshift__ (Int___init__ i) (Int___init__ j)) (Int___init__ (>> i j)) :ruleset array_api_ruleset) (function Int___invert__ (Int) Int) -(rewrite (Int___invert__ (Int___init__ __var__i)) (Int___init__ (not-i64 __var__i)) :ruleset array_api_ruleset) -(rewrite (Int_if_ (TRUE) __var__o __var__b) __var__o :ruleset array_api_ruleset) -(rewrite (Int_if_ (FALSE) __var__o __var__b) __var__b :ruleset array_api_ruleset) +(rewrite (Int___invert__ (Int___init__ i)) (Int___init__ (not-i64 i)) :ruleset array_api_ruleset) +(rewrite (Int_if_ (TRUE) o b) o :ruleset array_api_ruleset) +(rewrite (Int_if_ (FALSE) o b) b :ruleset array_api_ruleset) (function Boolean_bool (Boolean) bool) -(rule ((= __var__x (TRUE))) - ((set (Boolean_bool __var__x) true)) +(rule ((= x (TRUE))) + ((set (Boolean_bool x) true)) :ruleset array_api_ruleset ) -(rule ((= __var__x (FALSE))) - ((set (Boolean_bool __var__x) false)) +(rule ((= x (FALSE))) + ((set (Boolean_bool x) false)) :ruleset array_api_ruleset ) -(rewrite (Boolean___or__ (TRUE) __var__x) (TRUE) :ruleset array_api_ruleset) -(rewrite (Boolean___or__ (FALSE) __var__x) __var__x :ruleset array_api_ruleset) +(rewrite (Boolean___or__ (TRUE) x) (TRUE) :ruleset array_api_ruleset) +(rewrite (Boolean___or__ (FALSE) x) x :ruleset array_api_ruleset) (function Boolean___and__ (Boolean Boolean) Boolean) -(rewrite (Boolean___and__ (TRUE) __var__x) __var__x :ruleset array_api_ruleset) -(rewrite (Boolean___and__ (FALSE) __var__x) (FALSE) :ruleset array_api_ruleset) +(rewrite (Boolean___and__ (TRUE) x) x :ruleset array_api_ruleset) +(rewrite (Boolean___and__ (FALSE) x) (FALSE) :ruleset array_api_ruleset) (function Boolean_if_int (Boolean Int Int) Int) -(rewrite (Boolean_if_int (TRUE) __var__i __var__j) __var__i :ruleset array_api_ruleset) -(rewrite (Boolean_if_int (FALSE) __var__i __var__j) __var__j :ruleset array_api_ruleset) +(rewrite (Boolean_if_int (TRUE) i j) i :ruleset array_api_ruleset) +(rewrite (Boolean_if_int (FALSE) i j) j :ruleset array_api_ruleset) (function Boolean___invert__ (Boolean) Boolean) (rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) (rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) -(ruleset ruleset_5218390416) +(ruleset ruleset_5064589968) (function NDArray___eq__ (NDArray NDArray) NDArray) -(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse __var__x) (Int___init__ 1)) (NDArray_scalar (Value_int __var__i))) (NDArray___eq__ __var__x (NDArray_scalar (NDArray_index (unique_values __var__x) (TupleInt_from_vec (vec-of __var__i))))) :subsume :ruleset ruleset_5218390416) +(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 1)) (NDArray_scalar (Value_int i))) (NDArray___eq__ x (NDArray_scalar (NDArray_index (unique_values x) (TupleInt_from_vec (vec-of i))))) :subsume :ruleset ruleset_5064589968) (function count_values (NDArray NDArray) TupleValue :unextractable) -(rewrite (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) (NDArray_vector (count_values __var__x (unique_values __var__x))) :subsume :ruleset ruleset_5218390416) -(rewrite (count_values __var__x (NDArray_vector (TupleValue___add__ (TupleValue___init__ __var__v) __var__tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ __var__x (NDArray_scalar __var__v)) (OptionalIntOrTuple_none)))) (count_values __var__x (NDArray_vector __var__tv))) :subsume :ruleset ruleset_5218390416) -(rewrite (count_values __var__x (NDArray_vector (TupleValue___init__ __var__v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ __var__x (NDArray_scalar __var__v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_5218390416) +(rewrite (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (NDArray_vector (count_values x (unique_values x))) :subsume :ruleset ruleset_5064589968) +(rewrite (count_values x (NDArray_vector (TupleValue___add__ (TupleValue___init__ v) tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) (count_values x (NDArray_vector tv))) :subsume :ruleset ruleset_5064589968) +(rewrite (count_values x (NDArray_vector (TupleValue___init__ v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_5064589968) (function std (NDArray OptionalIntOrTuple) NDArray) (sort IntOrTuple) (function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) @@ -410,29 +1048,637 @@ (function ndarray-sqrt (NDArray) NDArray) (function mean (NDArray OptionalIntOrTuple Boolean) NDArray) (function square (NDArray) NDArray) -(rewrite (std __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i))) (ndarray-sqrt (mean (square (NDArray___sub__ __var__x (mean __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (FALSE))) :subsume :ruleset ruleset_5218390416) -(rewrite (mean __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (FALSE)) (NDArray___truediv__ (sum __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape __var__x) __var__i)))) :subsume :ruleset ruleset_5218390416) +(rewrite (std x (OptionalIntOrTuple_some (IntOrTuple_int i))) (ndarray-sqrt (mean (square (NDArray___sub__ x (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE))) :subsume :ruleset ruleset_5064589968) +(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) :subsume :ruleset ruleset_5064589968) (function expand_dims (NDArray Int) NDArray) -(rewrite (mean __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape __var__x) __var__i)))) __var__i) :subsume :ruleset ruleset_5218390416) -(function NDArray___matmul__ (NDArray NDArray) NDArray) +(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) i) :subsume :ruleset ruleset_5064589968) +(let %__expr_-4303590302573073911 (OptionalBool_none)) +(let %__expr_755392963466248965 (Int___init__ 2)) +(let %__expr_-990525301813695487 (OptionalDType_none)) +(let %__expr_8855076930384822898 (OptionalDevice_none)) (function NDArray_var (String) NDArray :cost 200) +(let %__expr_-8993155933335339035 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int %__expr_755392963466248965))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) +(let %__expr_-294044290527551890 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray %__expr_-8993155933335339035 %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) +(let %__expr_-277838035368029254 (OptionalInt_none)) +(let %__expr_-3636589015846229112 (Int___init__ 1)) +(let %__expr_-2409712210977886203 (IndexKey_int (Int___init__ 0))) +(let %__expr_-4671014753414284575 (Int___init__ 0)) +(let %__expr_-3193103899592592052 (Value_int (Int___init__ 1))) +(let %__expr_2090748637113203634 (Value_int %__expr_755392963466248965)) +(let %__expr_8800461850802719337 (Int___init__ 150)) +(let %__expr_-2515197891056208817 (Value_int (Int___init__ 0))) +(let %__expr_8968271287986727247 (unique_inverse %__expr_-8993155933335339035)) +(let %__expr_-346696523742155791 (TupleNDArray___getitem__ %__expr_8968271287986727247 %__expr_-3636589015846229112)) +(let %__expr_888968520189910301 (NDArray_scalar %__expr_-2515197891056208817)) +(let %__expr_8768355756772896527 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(sort MultiAxisIndexKeyItem) +(sort Slice) +(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) +(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) +(let %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 %__expr_-277838035368029254 %__expr_-277838035368029254))) +(let %__expr_-2454709417071966146 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) +(let %__expr_-3045893352967658126 (FALSE)) +(sort MultiAxisIndexKey) +(function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) +(sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) +(function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) +(function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) +(let %__expr_469855974343835084 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-4671014753414284575) %__expr_1503884865394975690)))) +(let %__expr_5553578895160412584 (Float___init__ 1.0)) +(let %__expr_-4976863396675597554 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-4671014753414284575))) (function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) (function zeros (TupleInt OptionalDType OptionalDevice) NDArray) (function OptionalDType_some (DType) OptionalDType) (sort Device) (function OptionalDevice_some (Device) OptionalDevice) (function NDArray_device (NDArray) Device) +(function IndexKey_ndarray (NDArray) IndexKey) +(let %__expr_6782829203688770802 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ %__expr_8968271287986727247 %__expr_-4671014753414284575)) %__expr_-4671014753414284575) (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-3636589015846229112))) (OptionalDType_some (NDArray_dtype %__expr_-2454709417071966146)) (OptionalDevice_some (NDArray_device %__expr_-2454709417071966146))) %__expr_469855974343835084 (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 %__expr_888968520189910301))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_-3193103899592592052)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_2090748637113203634)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126))) +(let %__expr_-8812288226011607914 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690)))) +(let %__expr_7373748767722314024 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690)))) +(function OptionalInt_some (Int) OptionalInt) +(let %__expr_-4845818903945609769 (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 %__expr_-2409712210977886203)))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575)) %__expr_-4976863396675597554)) +(let %__expr_-3351178202206693037 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 %__expr_-2409712210977886203)))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575))) +(function NDArray___mul__ (NDArray NDArray) NDArray) +(let %__expr_-3021532387541159817 (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_-277838035368029254)) +(let %__expr_5701152810033784528 (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575)) +(let %__expr_6487687809380785968 (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)) +(let %__expr_-6187783262847860806 (NDArray_dtype %__expr_-2454709417071966146)) +(let %__expr_-315001527591743086 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) +(let %__expr_-9037266425291862563 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126)) +(function NDArray___matmul__ (NDArray NDArray) NDArray) +(function IndexKey_slice (Slice) IndexKey) +(let %__expr_-6769231790204798359 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_5701152810033784528)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ %__expr_6487687809380785968 %__expr_-3636589015846229112))))))) (NDArray_T (NDArray___sub__ %__expr_6782829203688770802 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) +(let %__expr_1884788319774111777 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_5701152810033784528)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ %__expr_6487687809380785968 %__expr_-3636589015846229112))))))) (NDArray_T (NDArray___sub__ %__expr_6782829203688770802 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) %__expr_-3045893352967658126)) +(let %__expr_-5559224042962614522 (DType_int32)) +(let %__expr_-6676107104927312104 (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0))))) +(let %__expr_5888173354829332981 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802)) +(let %__expr_2473377876388853102 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) +(let %__expr_-3035386490804382424 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) +(simplify (saturate (seq (run array_api_ruleset) (run ruleset_5064589968))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-2454709417071966146 %__expr_5888173354829332981) (NDArray___matmul__ %__expr_-3035386490804382424 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_1884788319774111777 %__expr_755392963466248965)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_-6769231790204798359 (NDArray___mul__ (NDArray_scalar (Value_float (Float___init__ 0.0001))) (NDArray___getitem__ %__expr_-6769231790204798359 %__expr_-2409712210977886203))) %__expr_-5559224042962614522) (OptionalIntOrTuple_none))))) %__expr_-277838035368029254)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some %__expr_755392963466248965) %__expr_-277838035368029254))))))) +(pop 1) +(sort Int) +(function Int___init__ (i64) Int) +(let %__expr_-4671014753414284575 (Int___init__ 0)) +(let %__expr_8800461850802719337 (Int___init__ 150)) +(sort NDArray) +(sort IndexKey) +(function NDArray___getitem__ (NDArray IndexKey) NDArray) +(function assume_isfinite (NDArray) NDArray) +(sort TupleInt) +(function assume_shape (NDArray TupleInt) NDArray) +(sort DType) +(function assume_dtype (NDArray DType) NDArray) +(function NDArray_var (String) NDArray :cost 200) +(function DType_float64 () DType) +(sort Vec_Int (Vec Int)) +(function TupleInt_from_vec (Vec_Int) TupleInt) +(function IndexKey_ndarray (NDArray) IndexKey) +(function NDArray___eq__ (NDArray NDArray) NDArray) +(sort TupleValue) +(function assume_value_one_of (NDArray TupleValue) NDArray) +(function DType_int64 () DType) +(function TupleValue___add__ (TupleValue TupleValue) TupleValue) +(sort Value) +(function TupleValue___init__ (Value) TupleValue) +(function Value_int (Int) Value) +(function NDArray_scalar (Value) NDArray) +(let %__expr_-3473821386163038220 (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int %__expr_-4671014753414284575)))))) +(let %__expr_-3193103899592592052 (Value_int (Int___init__ 1))) +(let %__expr_2090748637113203634 (Value_int (Int___init__ 2))) +(let %__expr_-4688800770552726910 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4)))))) +(let %__expr_8326680938997940725 (NDArray___getitem__ %__expr_-4688800770552726910 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-3193103899592592052))))) +(sort OptionalIntOrTuple) +(sort IntOrTuple) +(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) +(function IntOrTuple_int (Int) IntOrTuple) +(let %__expr_-4976863396675597554 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-4671014753414284575))) +(let %__expr_-3636589015846229112 (Int___init__ 1)) +(let %__expr_-2298714861395141926 (Int___init__ 4)) +(let %__expr_-2515197891056208817 (Value_int %__expr_-4671014753414284575)) +(let %__expr_4101187900440652363 (DType_float64)) +(sort MultiAxisIndexKeyItem) +(sort Slice) +(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) +(sort OptionalInt) +(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) +(function OptionalInt_none () OptionalInt) +(let %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) +(let %__expr_5698022980052622903 (NDArray___getitem__ %__expr_-4688800770552726910 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_2090748637113203634))))) (sort MultiAxisIndexKey) (function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) -(sort MultiAxisIndexKeyItem) (sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) (function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) (function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) -(sort Slice) -(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) -(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) -(function IndexKey_ndarray (NDArray) IndexKey) -(function NDArray___mul__ (NDArray NDArray) NDArray) +(let %__expr_-795645184301196957 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) %__expr_1503884865394975690)))) +(let %__expr_3323365144993085153 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690)))) +(let %__expr_7232724363220639299 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-4671014753414284575) %__expr_1503884865394975690)))) +(let %__expr_755392963466248965 (Int___init__ 2)) +(function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) +(sort OptionalDType) +(sort OptionalDevice) +(function zeros (TupleInt OptionalDType OptionalDevice) NDArray) +(function OptionalDType_some (DType) OptionalDType) +(sort Device) +(function OptionalDevice_some (Device) OptionalDevice) +(function NDArray_device (NDArray) Device) +(function NDArray___truediv__ (NDArray NDArray) NDArray) +(function sum (NDArray OptionalIntOrTuple) NDArray) +(function TupleInt___getitem__ (TupleInt Int) Int) +(function NDArray_shape (NDArray) TupleInt) +(let %__expr_-4084543222422775141 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) %__expr_-2298714861395141926)) (OptionalDType_some %__expr_4101187900440652363) (OptionalDevice_some (NDArray_device %__expr_-4688800770552726910))) %__expr_7232724363220639299 (NDArray___truediv__ (sum %__expr_-3473821386163038220 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-3473821386163038220) %__expr_-4671014753414284575))))) %__expr_3323365144993085153 (NDArray___truediv__ (sum %__expr_8326680938997940725 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_8326680938997940725) %__expr_-4671014753414284575))))) %__expr_-795645184301196957 (NDArray___truediv__ (sum %__expr_5698022980052622903 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_5698022980052622903) %__expr_-4671014753414284575)))))) +(sort TupleNDArray) +(function concat (TupleNDArray OptionalInt) NDArray) +(function TupleNDArray___add__ (TupleNDArray TupleNDArray) TupleNDArray) +(function TupleNDArray___init__ (NDArray) TupleNDArray) +(function NDArray___sub__ (NDArray NDArray) NDArray) (function OptionalInt_some (Int) OptionalInt) +(let %__expr_-5454083913435783283 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-3473821386163038220 (NDArray___getitem__ %__expr_-4084543222422775141 %__expr_7232724363220639299))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_8326680938997940725 (NDArray___getitem__ %__expr_-4084543222422775141 %__expr_3323365144993085153))) (TupleNDArray___init__ (NDArray___sub__ %__expr_5698022980052622903 (NDArray___getitem__ %__expr_-4084543222422775141 %__expr_-795645184301196957))))) (OptionalInt_some %__expr_-4671014753414284575))) +(let %__expr_4590898867152193220 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_2090748637113203634))) +(function OptionalIntOrTuple_none () OptionalIntOrTuple) +(let %__expr_2665112130965315478 (OptionalIntOrTuple_none)) +(let %__expr_-608551224937479330 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-3193103899592592052))) +(let %__expr_-5257952192217027907 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int %__expr_-4671014753414284575)))) +(let %__expr_8511786851084399467 (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2))))))) +(function square (NDArray) NDArray) +(function expand_dims (NDArray Int) NDArray) +(let %__expr_-1359368604848933744 (square (NDArray___sub__ %__expr_-5454083913435783283 (expand_dims (NDArray___truediv__ (sum %__expr_-5454083913435783283 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-5454083913435783283) %__expr_-4671014753414284575)))) %__expr_-4671014753414284575)))) +(let %__expr_-277838035368029254 (OptionalInt_none)) +(let %__expr_888968520189910301 (NDArray_scalar (Value_int %__expr_-4671014753414284575))) +(let %__expr_-3240316658536798486 1) +(function ndarray-sqrt (NDArray) NDArray) +(let %__expr_-2066297328423310743 (ndarray-sqrt (NDArray___truediv__ (sum %__expr_-1359368604848933744 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-1359368604848933744) %__expr_-4671014753414284575)))))) +(function Value_to_int (Value) Int) +(function NDArray_to_value (NDArray) Value) +(function astype (NDArray DType) NDArray) +(function NDArray___gt__ (NDArray NDArray) NDArray) +(function TupleNDArray___getitem__ (TupleNDArray Int) NDArray) +(sort Boolean) +(function svd (NDArray Boolean) TupleNDArray) +(function NDArray___mul__ (NDArray NDArray) NDArray) +(sort Float) +(function Value_float (Float) Value) +(function Float_rational (Rational) Float :cost 2) +(function FALSE () Boolean) +(function Float___init__ (f64) Float :cost 3) +(function DType_int32 () DType) +(let %__expr_4399334460195737951 (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 147))))) (NDArray___truediv__ %__expr_-5454083913435783283 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486))))))) (FALSE)) %__expr_-3636589015846229112) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) %__expr_2665112130965315478)))) %__expr_-277838035368029254)) +(let %__expr_-6024165510290428751 150) +(function NDArray_vector (TupleValue) NDArray) +(let %__expr_-4005821647467750755 (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-5257952192217027907 %__expr_2665112130965315478))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-608551224937479330 %__expr_2665112130965315478))) (TupleValue___init__ (NDArray_to_value (sum %__expr_4590898867152193220 %__expr_2665112130965315478)))))) %__expr_4101187900440652363) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-6024165510290428751 %__expr_-3240316658536798486)))))) +(let %__expr_-3931088046849885566 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 147))))) (NDArray___truediv__ %__expr_-5454083913435783283 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486))))))) (FALSE)) %__expr_-3636589015846229112)) +(let %__expr_-2315365344465410478 (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 147))))) (NDArray___truediv__ %__expr_-5454083913435783283 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486))))))) (FALSE))) +(let %__expr_-512142818984394169 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486)))))) +(let %__expr_-3045893352967658126 (FALSE)) +(function NDArray___matmul__ (NDArray NDArray) NDArray) +(function NDArray_T (NDArray) NDArray) (function IndexKey_slice (Slice) IndexKey) -(simplify (saturate (seq (run array_api_ruleset) (run ruleset_5218390416))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))))) (NDArray___matmul__ (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))))) (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0)) (Int___init__ 1)))))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE)))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none)))))) (FALSE)) (Int___init__ 2))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0)) (Int___init__ 1)))))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE)))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none)))))) (FALSE)) (Int___init__ 1)) (NDArray___mul__ (NDArray_scalar (Value_float (Float___init__ 0.0001))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0)) (Int___init__ 1)))))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE)))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none)))))) (FALSE)) (Int___init__ 1)) (IndexKey_int (Int___init__ 0))))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Int___init__ 2)) (OptionalInt_none)))))))) +(let %__expr_7487275443900416961 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_8800461850802719337)) %__expr_-4005821647467750755) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 2)))))) (NDArray_T (NDArray___sub__ %__expr_-4084543222422775141 (NDArray___matmul__ %__expr_-4005821647467750755 %__expr_-4084543222422775141))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-2315365344465410478 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_4399334460195737951) %__expr_1503884865394975690)))) %__expr_-512142818984394169)) (NDArray___getitem__ %__expr_-3931088046849885566 (IndexKey_slice %__expr_4399334460195737951)))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) +(let %__expr_6183362166771178405 2) +(let %__expr_-5395166694618893324 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_8800461850802719337)) %__expr_-4005821647467750755) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 2)))))) (NDArray_T (NDArray___sub__ %__expr_-4084543222422775141 (NDArray___matmul__ %__expr_-4005821647467750755 %__expr_-4084543222422775141))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-2315365344465410478 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_4399334460195737951) %__expr_1503884865394975690)))) %__expr_-512142818984394169)) (NDArray___getitem__ %__expr_-3931088046849885566 (IndexKey_slice %__expr_4399334460195737951)))) %__expr_-3045893352967658126)) +(let %__expr_-5559224042962614522 (DType_int32)) +(let %__expr_2976592046696346842 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) +(let %__expr_-7009371984517665352 (NDArray___matmul__ %__expr_-4005821647467750755 %__expr_-4084543222422775141)) +(let %__expr_-6216319311343226647 (NDArray_var "y")) +(let %__expr_6240449845396421003 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-2315365344465410478 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_4399334460195737951) %__expr_1503884865394975690)))) %__expr_-512142818984394169)) (NDArray___getitem__ %__expr_-3931088046849885566 (IndexKey_slice %__expr_4399334460195737951)))) +(let %__expr_4033771585938608070 (NDArray_var "X")) +(sort Program) +(function ndarray_function_two (NDArray NDArray NDArray) Program) +(function IndexKey_int (Int) IndexKey) +(let %fn_program (ndarray_function_two (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-4688800770552726910 %__expr_-7009371984517665352) (NDArray___matmul__ %__expr_6240449845396421003 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-5395166694618893324 %__expr_755392963466248965)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_7487275443900416961 (NDArray___mul__ %__expr_2976592046696346842 (NDArray___getitem__ %__expr_7487275443900416961 (IndexKey_int %__expr_-4671014753414284575)))) %__expr_-5559224042962614522) %__expr_2665112130965315478)))) %__expr_-277838035368029254)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some %__expr_755392963466248965) %__expr_-277838035368029254)))))) %__expr_4033771585938608070 %__expr_-6216319311343226647)) +(ruleset ruleset_5064591888) +(function ndarray_program (NDArray) Program) +(function Program___init__ (String bool) Program) +(rewrite (ndarray_program (NDArray_var s)) (Program___init__ s true) :ruleset ruleset_5064591888) +(function Program_statement (Program Program) Program :unextractable) +(function Program___add__ (Program Program) Program) +(function dtype_program (DType) Program) +(rewrite (ndarray_program (assume_dtype z dtype)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".dtype == " false)) (dtype_program dtype))) :ruleset ruleset_5064591888) +(function tuple_int_program (TupleInt) Program) +(rewrite (ndarray_program (assume_shape z ti)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".shape == " false)) (tuple_int_program ti))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (assume_isfinite z)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___init__ "assert np.all(np.isfinite(" false) (ndarray_program z)) (Program___init__ "))" false))) :ruleset ruleset_5064591888) +(function tuple_value_program (TupleValue) Program) +(rewrite (ndarray_program (assume_value_one_of z tv)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert set(np.unique(" false) (ndarray_program z)) (Program___init__ ")) == set(" false)) (tuple_value_program tv)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(sort OptionalBool) +(function reshape (NDArray TupleInt OptionalBool) NDArray) +(function Program_assign (Program) Program) +(rewrite (ndarray_program (reshape y ti ob)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".reshape(" false)) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (astype y dtype)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".astype(" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function tuple_ndarray_program (TupleNDArray) Program) +(function unique_counts (NDArray) TupleNDArray) +(rewrite (tuple_ndarray_program (unique_counts x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_counts=True)" false))) :ruleset ruleset_5064591888) +(function unique_inverse (NDArray) TupleNDArray) +(rewrite (tuple_ndarray_program (unique_inverse x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_inverse=True)" false))) :ruleset ruleset_5064591888) +(function int_program (Int) Program) +(rewrite (ndarray_program (TupleNDArray___getitem__ tnd i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program tnd) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(function value_program (Value) Program) +(rewrite (ndarray_program (NDArray_scalar v)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (value_program v)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function OptionalDType_none () OptionalDType) +(rewrite (ndarray_program (zeros ti (OptionalDType_none) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (zeros ti (OptionalDType_some dtype) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ", dtype=" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function unique_values (NDArray) NDArray) +(rewrite (ndarray_program (unique_values x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function NDArray___add__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___add__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " + " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___sub__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " - " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___mul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " * " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___truediv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " / " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___lt__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___lt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " < " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___le__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___le__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " <= " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___gt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " > " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___ge__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___ge__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >= " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___eq__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " == " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___matmul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " @ " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___mod__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___mod__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " % " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___and__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___and__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " & " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___or__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___or__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " | " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___xor__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___xor__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ^ " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___lshift__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___lshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " << " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___rshift__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___rshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >> " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___floordiv__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___floordiv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " // " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function NDArray___pow__ (NDArray NDArray) NDArray) +(rewrite (ndarray_program (NDArray___pow__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ** " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(function index_key_program (IndexKey) Program) +(rewrite (ndarray_program (NDArray___setitem__ x idx y)) (Program_statement (Program_assign (ndarray_program x)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program_assign (ndarray_program x)) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "] = " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___getitem__ x idx)) (Program___add__ (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (square x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.square(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (expand_dims x i)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.expand_dims(" false) (ndarray_program x)) (Program___init__ ", " false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function mean (NDArray OptionalIntOrTuple Boolean) NDArray) +(rewrite (ndarray_program (mean x %__expr_2665112130965315478 %__expr_-3045893352967658126)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function int_or_tuple_program (IntOrTuple) Program) +(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) %__expr_-3045893352967658126)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function TRUE () Boolean) +(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ", keepdims=True)" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (concat tnd %__expr_-277838035368029254)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (concat tnd (OptionalInt_some i))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ", axis=" false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray_vector tv)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (tuple_value_program tv)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function std (NDArray OptionalIntOrTuple) NDArray) +(rewrite (ndarray_program (std x %__expr_2665112130965315478)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (std x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program (svd x (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program (svd x %__expr_-3045893352967658126)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ", full_matrices=False)" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (ndarray-sqrt x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sqrt(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray_T x)) (Program___add__ (ndarray_program x) (Program___init__ ".T" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (sum x %__expr_2665112130965315478)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (sum x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program (NDArray_shape x)) (Program___add__ (ndarray_program x) (Program___init__ ".shape" false)) :ruleset ruleset_5064591888) +(function ndarray-abs (NDArray) NDArray) +(rewrite (ndarray_program (ndarray-abs x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(function optional_int_or_tuple_program (OptionalIntOrTuple) Program) +(rewrite (optional_int_or_tuple_program (OptionalIntOrTuple_some it)) (int_or_tuple_program it) :ruleset ruleset_5064591888) +(rewrite (optional_int_or_tuple_program %__expr_2665112130965315478) (Program___init__ "None" false) :ruleset ruleset_5064591888) +(rewrite (int_or_tuple_program (IntOrTuple_int x)) (int_program x) :ruleset ruleset_5064591888) +(function IntOrTuple_tuple (TupleInt) IntOrTuple) +(rewrite (int_or_tuple_program (IntOrTuple_tuple t)) (tuple_int_program t) :ruleset ruleset_5064591888) +(function IndexKey_ELLIPSIS () IndexKey) +(rewrite (index_key_program (IndexKey_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_5064591888) +(rewrite (index_key_program (IndexKey_int i)) (int_program i) :ruleset ruleset_5064591888) +(function slice_program (Slice) Program) +(rewrite (index_key_program (IndexKey_slice s)) (slice_program s) :ruleset ruleset_5064591888) +(function multi_axis_index_key_program (MultiAxisIndexKey) Program) +(rewrite (index_key_program (IndexKey_multi_axis key)) (multi_axis_index_key_program key) :ruleset ruleset_5064591888) +(rewrite (index_key_program (IndexKey_ndarray a)) (ndarray_program a) :ruleset ruleset_5064591888) +(sort UnstableFn_MultiAxisIndexKeyItem_Int (UnstableFn (Int) MultiAxisIndexKeyItem)) +(function Int___add__ (Int Int) Int) +(function idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ (UnstableFn_MultiAxisIndexKeyItem_Int Int) MultiAxisIndexKeyItem) +(rewrite (idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_-3636589015846229112)) :ruleset ruleset_5064591888) +(function MultiAxisIndexKey___init__ (Int UnstableFn_MultiAxisIndexKeyItem_Int) MultiAxisIndexKey) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ %__expr_-4671014753414284575 idx_fn)) (Program___init__ "" false) :ruleset ruleset_5064591888) +(function multi_axis_index_key_item_program (MultiAxisIndexKeyItem) Program) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (unstable-app idx_fn %__expr_-4671014753414284575)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ (- k %__expr_-3240316658536798486)) (unstable-fn "idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ "," false)) :when ((= (vec-length vec) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-remove vec 0)))) :when ((> (vec-length vec) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_int i)) (int_program i) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_slice s)) (slice_program s) :ruleset ruleset_5064591888) +(function MultiAxisIndexKeyItem_ELLIPSIS () MultiAxisIndexKeyItem) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_5064591888) +(function MultiAxisIndexKeyItem_NONE () MultiAxisIndexKeyItem) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_NONE)) (Program___init__ "None" false) :ruleset ruleset_5064591888) +(function optional_int_slice_program (OptionalInt) Program) +(rewrite (slice_program (Slice___init__ start stop %__expr_-277838035368029254)) (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) :ruleset ruleset_5064591888) +(rewrite (slice_program (Slice___init__ start stop (OptionalInt_some i))) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) (Program___init__ ":" false)) (int_program i)) :ruleset ruleset_5064591888) +(rewrite (optional_int_slice_program %__expr_-277838035368029254) (Program___init__ "" false) :ruleset ruleset_5064591888) +(rewrite (optional_int_slice_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_5064591888) +(function optional_int_program (OptionalInt) Program) +(rewrite (optional_int_program %__expr_-277838035368029254) (Program___init__ "None" false) :ruleset ruleset_5064591888) +(rewrite (optional_int_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_5064591888) +(function optional_dtype_program (OptionalDType) Program) +(rewrite (optional_dtype_program (OptionalDType_none)) (Program___init__ "None" false) :ruleset ruleset_5064591888) +(rewrite (optional_dtype_program (OptionalDType_some dtype)) (dtype_program dtype) :ruleset ruleset_5064591888) +(function tuple_ndarray_program_inner (TupleNDArray) Program) +(rewrite (tuple_ndarray_program r) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_ndarray_program_inner r)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program_inner (TupleNDArray___init__ x)) (Program___add__ (ndarray_program x) (Program___init__ "," false)) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program_inner (TupleNDArray___add__ l r)) (Program___add__ (Program___add__ (tuple_ndarray_program_inner l) (Program___init__ " " false)) (tuple_ndarray_program_inner r)) :ruleset ruleset_5064591888) +(function TupleNDArray_length (TupleNDArray) Int) +(rewrite (int_program (TupleNDArray_length l)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_ndarray_program l)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (TupleNDArray___getitem__ l i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program l) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(function tuple_value_program_inner (TupleValue) Program) +(rewrite (tuple_value_program tv1) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_value_program_inner tv1)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (tuple_value_program_inner (TupleValue___add__ tv1 tv2)) (Program___add__ (Program___add__ (tuple_value_program_inner tv1) (Program___init__ " " false)) (tuple_value_program_inner tv2)) :ruleset ruleset_5064591888) +(rewrite (tuple_value_program_inner (TupleValue___init__ v)) (Program___add__ (value_program v) (Program___init__ "," false)) :ruleset ruleset_5064591888) +(rewrite (value_program (Value_int i)) (int_program i) :ruleset ruleset_5064591888) +(function Value_bool (Boolean) Value) +(function bool_program (Boolean) Program) +(rewrite (value_program (Value_bool b)) (bool_program b) :ruleset ruleset_5064591888) +(function float_program (Float) Program) +(rewrite (value_program (Value_float f)) (float_program f) :ruleset ruleset_5064591888) +(rewrite (value_program (NDArray_to_value x)) (ndarray_program x) :ruleset ruleset_5064591888) +(function Value___lt__ (Value Value) Value) +(rewrite (value_program (Value___lt__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " < " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Value___truediv__ (Value Value) Value) +(rewrite (value_program (Value___truediv__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " / " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Value_to_bool (Value) Boolean) +(rewrite (bool_program (Value_to_bool v1)) (value_program v1) :ruleset ruleset_5064591888) +(rewrite (int_program (Value_to_int v1)) (value_program v1) :ruleset ruleset_5064591888) +(rewrite (float_program (Float___init__ f64_)) (Program___init__ (to-string f64_) false) :ruleset ruleset_5064591888) +(function Float_abs (Float) Float) +(rewrite (float_program (Float_abs f)) (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (float_program f)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Float_from_int (Int) Float) +(rewrite (float_program (Float_from_int i)) (int_program i) :ruleset ruleset_5064591888) +(function Float___add__ (Float Float) Float) +(rewrite (float_program (Float___add__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " + " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Float___sub__ (Float Float) Float) +(rewrite (float_program (Float___sub__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " - " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Float___mul__ (Float Float) Float) +(rewrite (float_program (Float___mul__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " * " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Float___truediv__ (Float Float) Float) +(rewrite (float_program (Float___truediv__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " / " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ " / " false)) (Program___init__ (to-string (denom r)) false)) (Program___init__ ")" false)) :when ((!= (denom r) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) +(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ ")" false)) :when ((= (denom r) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) +(rewrite (dtype_program %__expr_4101187900440652363) (Program___init__ "np.dtype(np.float64)" false) :ruleset ruleset_5064591888) +(function DType_float32 () DType) +(rewrite (dtype_program (DType_float32)) (Program___init__ "np.dtype(np.float32)" false) :ruleset ruleset_5064591888) +(rewrite (dtype_program (DType_int64)) (Program___init__ "np.dtype(np.int64)" false) :ruleset ruleset_5064591888) +(rewrite (dtype_program %__expr_-5559224042962614522) (Program___init__ "np.dtype(np.int32)" false) :ruleset ruleset_5064591888) +(function DType_bool () DType) +(rewrite (dtype_program (DType_bool)) (Program___init__ "np.dtype(np.bool)" false) :ruleset ruleset_5064591888) +(function DType_object () DType) +(rewrite (dtype_program (DType_object)) (Program___init__ "np.dtype(np.object_)" false) :ruleset ruleset_5064591888) +(function Program_function_two (Program Program Program String) Program) +(sort UnstableFn_Int_Int (UnstableFn (Int) Int)) +(function idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ (UnstableFn_Int_Int Int) Int) +(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_-3636589015846229112)) :ruleset ruleset_5064591888) +(rewrite (int_program (TupleInt___getitem__ ti i)) (Program___add__ (Program___add__ (Program___add__ (tuple_int_program ti) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(function TupleInt_length (TupleInt) Int) +(rewrite (int_program (TupleInt_length ti)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_int_program ti)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function tuple_int_program_inner (TupleInt) Program) +(rewrite (tuple_int_program ti) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_int_program_inner ti)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt :unextractable) +(rewrite (tuple_int_program_inner (TupleInt___init__ %__expr_-4671014753414284575 idx_fn)) (Program___init__ "" false) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program_inner (TupleInt___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (int_program (unstable-app idx_fn %__expr_-4671014753414284575)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt___init__ (Int___init__ (- k %__expr_-3240316658536798486)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program_inner (TupleInt_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt_from_vec (vec-remove vec_int 0)))) :when ((> (vec-length vec_int) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ "," false)) :when ((= (vec-length vec_int) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___init__ i64_)) (Program___init__ (to-string i64_) false) :ruleset ruleset_5064591888) +(function Int___invert__ (Int) Int) +(rewrite (int_program (Int___invert__ i)) (Program___add__ (Program___init__ "~" false) (int_program i)) :ruleset ruleset_5064591888) +(function Int___lt__ (Int Int) Boolean) +(rewrite (bool_program (Int___lt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " < " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___le__ (Int Int) Boolean) +(rewrite (bool_program (Int___le__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " <= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___gt__ (Int Int) Boolean) +(rewrite (bool_program (Int___gt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " > " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___ge__ (Int Int) Boolean) +(rewrite (bool_program (Int___ge__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___eq__ (Int Int) Boolean) +(rewrite (bool_program (Int___eq__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " == " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___add__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " + " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___sub__ (Int Int) Int) +(rewrite (int_program (Int___sub__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " - " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___mul__ (Int Int) Int) +(rewrite (int_program (Int___mul__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " * " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___truediv__ (Int Int) Int) +(rewrite (int_program (Int___truediv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " / " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___mod__ (Int Int) Int) +(rewrite (int_program (Int___mod__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " % " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___pow__ (Int Int) Int) +(rewrite (int_program (Int___pow__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ** " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___and__ (Int Int) Int) +(rewrite (int_program (Int___and__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " & " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___or__ (Int Int) Int) +(rewrite (int_program (Int___or__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " | " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___xor__ (Int Int) Int) +(rewrite (int_program (Int___xor__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ^ " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___lshift__ (Int Int) Int) +(rewrite (int_program (Int___lshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " << " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___rshift__ (Int Int) Int) +(rewrite (int_program (Int___rshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >> " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(function Int___floordiv__ (Int Int) Int) +(rewrite (int_program (Int___floordiv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " // " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (bool_program (TRUE)) (Program___init__ "True" false) :ruleset ruleset_5064591888) +(rewrite (bool_program %__expr_-3045893352967658126) (Program___init__ "False" false) :ruleset ruleset_5064591888) +(ruleset ruleset_5063997584) +(function Program_expr (Program) String) +(function Program_statements (Program) String) +(function Program_next_sym (Program) i64) +(function Program_is_identifer (Program) bool) +(function Program_compile (Program i64) Unit :default ()) +(rule ((= p (Program___init__ s b)) + (Program_compile p i)) + ((set (Program_expr p) s) + (set (Program_statements p) "") + (set (Program_next_sym p) i) + (set (Program_is_identifer p) b)) + :ruleset ruleset_5063997584 ) +(function Program_expr_to_statement (Program) Program) +(rewrite (Program_statement p1 p2) (Program___add__ p1 (Program_expr_to_statement p2)) :ruleset ruleset_5063997584) +(function Program_parent (Program) Program :unextractable :merge old) +(rule ((= p (Program_expr_to_statement p1)) + (Program_compile p i)) + ((set (Program_parent p1) p) + (set (Program_is_identifer p) false)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_expr_to_statement p1)) + (Program_compile p i) + (= (Program_parent p1) p)) + ((Program_compile p1 i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_expr_to_statement p1)) + (Program_compile p i) + (!= (Program_parent p1) p) + (= s1 (Program_expr p1))) + ((set (Program_statements p) (+ s1 " +")) + (set (Program_next_sym p) i) + (set (Program_expr p) "")) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_expr_to_statement p1)) + (= (Program_parent p1) p) + (= s1 (Program_expr p1)) + (= s2 (Program_statements p1)) + (= i (Program_next_sym p1))) + ((set (Program_statements p) (+ s2 s1 " +")) + (set (Program_next_sym p) i) + (set (Program_expr p) "")) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (= (Program_expr p) (Program_expr p1)) + (= b (Program_is_identifer p1))) + ((set (Program_is_identifer p) b)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (= (Program_expr p) (Program_expr p2)) + (= b (Program_is_identifer p2))) + ((set (Program_is_identifer p) b)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (!= (Program_expr p) (Program_expr p1)) + (!= (Program_expr p) (Program_expr p2))) + ((set (Program_is_identifer p) false)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (Program_compile p i)) + ((set (Program_parent p1) p)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (Program_compile p i) + (= (Program_parent p1) p)) + ((Program_compile p1 i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (Program_compile p i) + (Program_next_sym p1)) + ((set (Program_parent p2) p)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (Program_compile p i) + (!= (Program_parent p1) p) + (= (Program_parent p2) p)) + ((Program_compile p2 i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (Program_compile p i2) + (= (Program_parent p1) p) + (= i (Program_next_sym p1)) + (= (Program_parent p2) p)) + ((Program_compile p2 i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (= s1 (Program_expr p1)) + (= s2 (Program_expr p2))) + ((set (Program_expr p) (+ s1 s2))) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (= (Program_parent p1) p) + (= (Program_parent p2) p) + (= s1 (Program_statements p1)) + (= s2 (Program_statements p2)) + (= i (Program_next_sym p2))) + ((set (Program_statements p) (+ s1 s2)) + (set (Program_next_sym p) i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (Program_compile p i) + (!= (Program_parent p1) p) + (!= (Program_parent p2) p)) + ((set (Program_statements p) "") + (set (Program_next_sym p) i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (= (Program_parent p1) p) + (!= (Program_parent p2) p) + (= s1 (Program_statements p1)) + (= i (Program_next_sym p1))) + ((set (Program_statements p) s1) + (set (Program_next_sym p) i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program___add__ p1 p2)) + (= (Program_parent p2) p) + (!= (Program_parent p1) p) + (= s2 (Program_statements p2)) + (= i (Program_next_sym p2))) + ((set (Program_statements p) s2) + (set (Program_next_sym p) i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_assign p1)) + (Program_compile p i)) + ((set (Program_parent p1) p) + (set (Program_is_identifer p) true)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_assign p1)) + (Program_compile p i) + (= (Program_parent p1) p)) + ((Program_compile p1 i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_assign p1)) + (= (Program_parent p1) p) + (= s1 (Program_statements p1)) + (= i (Program_next_sym p1)) + (= s2 (Program_expr p1)) + (= (Program_is_identifer p1) false)) + ((set (Program_statements p) (+ s1 (+ "_" (to-string i)) " = " s2 " +")) + (set (Program_expr p) (+ "_" (to-string i))) + (set (Program_next_sym p) (+ i %__expr_-3240316658536798486))) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_assign p1)) + (!= (Program_parent p1) p) + (Program_compile p i) + (= s2 (Program_expr p1)) + (= (Program_is_identifer p1) false)) + ((set (Program_statements p) (+ (+ "_" (to-string i)) " = " s2 " +")) + (set (Program_expr p) (+ "_" (to-string i))) + (set (Program_next_sym p) (+ i %__expr_-3240316658536798486))) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_assign p1)) + (= (Program_parent p1) p) + (= s1 (Program_statements p1)) + (= i (Program_next_sym p1)) + (= s2 (Program_expr p1)) + (= (Program_is_identifer p1) true)) + ((set (Program_statements p) s1) + (set (Program_expr p) s2) + (set (Program_next_sym p) i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_assign p1)) + (!= (Program_parent p1) p) + (Program_compile p i) + (= s2 (Program_expr p1)) + (= (Program_is_identifer p1) true)) + ((set (Program_statements p) "") + (set (Program_expr p) s2) + (set (Program_next_sym p) i)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_function_two p1 p2 p3 s1)) + (Program_compile p i)) + ((set (Program_parent p2) p) + (set (Program_parent p3) p) + (set (Program_parent p1) p) + (Program_compile p2 i) + (Program_compile p3 i) + (Program_compile p1 i) + (set (Program_is_identifer p) true)) + :ruleset ruleset_5063997584 ) +(rule ((= p (Program_function_two p1 p2 p3 s1)) + (Program_compile p i) + (= s2 (Program_expr p1)) + (= s3 (Program_statements p1)) + (= s4 (Program_expr p2)) + (= s5 (Program_expr p3))) + ((set (Program_statements p) (+ "def " s1 "(" s4 ", " s5 "): + " (replace s3 " +" " + ") "return " s2 " +")) + (set (Program_next_sym p) i) + (set (Program_expr p) s1)) + :ruleset ruleset_5063997584 ) +(rule ((= f (ndarray_function_two res l r))) + ((union f (Program_function_two (ndarray_program res) (ndarray_program l) (ndarray_program r) "__fn")) + (Program_compile f 0)) + :ruleset ruleset_5064591888 ) +(run-schedule (seq (saturate (run ruleset_5064591888)) (saturate (run ruleset_5063997584)))) +(extract (Program_statements %fn_program) 0) From 710d7a24c0fb1000dce969c79e5589188148e807 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 17 Oct 2024 16:47:32 -0400 Subject: [PATCH 02/11] Add extracts --- tests/python_array_optimize.egg | 858 +++++++++++++++++--------------- 1 file changed, 457 insertions(+), 401 deletions(-) diff --git a/tests/python_array_optimize.egg b/tests/python_array_optimize.egg index a24745c2..52f9e1e3 100644 --- a/tests/python_array_optimize.egg +++ b/tests/python_array_optimize.egg @@ -1,10 +1,9 @@ ;; created by running test_source_optimized and removing the PyObject functions -;; pytest -s python/tests/test_array_api.py::TestLDA::test_source_optimized --benchmark-disable (push 1) (push 1) (sort DType) (function DType_float64 () DType) -(let %__expr_4101187900440652363 (DType_float64)) +(let %__expr_-458881301673629543 (DType_float64)) (sort Boolean) (function DType___eq__ (DType DType) Boolean) (sort NDArray) @@ -18,7 +17,7 @@ (sort Vec_Int (Vec Int)) (function TupleInt_from_vec (Vec_Int) TupleInt) (function Int___init__ (i64) Int) -(DType___eq__ %__expr_4101187900440652363 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) +(DType___eq__ %__expr_-458881301673629543 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) (ruleset array_api_ruleset) (function i-Int__-Int_i (Int Int) Int) (rewrite (i-Int__-Int_i i _) i :ruleset array_api_ruleset) @@ -206,7 +205,7 @@ (rewrite (sum (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_int (NDArray_size x))) :ruleset array_api_ruleset) (rewrite (sum (astype (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) dtype) (OptionalIntOrTuple_none)) (astype (NDArray_scalar (Value_int (NDArray_size x))) dtype) :ruleset array_api_ruleset) (rewrite (NDArray_dtype (astype x dtype)) dtype :ruleset array_api_ruleset) -(rewrite (astype (NDArray_scalar (Value_int (Int___init__ i))) %__expr_4101187900440652363) (NDArray_scalar (Value_float (Float___init__ (to-f64 i)))) :ruleset array_api_ruleset) +(rewrite (astype (NDArray_scalar (Value_int (Int___init__ i))) %__expr_-458881301673629543) (NDArray_scalar (Value_float (Float___init__ (to-f64 i)))) :ruleset array_api_ruleset) (sort OptionalInt) (function concat (TupleNDArray OptionalInt) NDArray) (function TupleNDArray___init__ (NDArray) TupleNDArray) @@ -261,7 +260,7 @@ (rewrite (TupleValue_includes (TupleValue___add__ ti ti2) v) (Boolean___or__ (TupleValue_includes ti v) (TupleValue_includes ti2 v)) :ruleset array_api_ruleset) (function DType_int64 () DType) (rewrite (Value_dtype (Value_int i)) (DType_int64) :ruleset array_api_ruleset) -(rewrite (Value_dtype (Value_float f)) %__expr_4101187900440652363 :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_float f)) %__expr_-458881301673629543 :ruleset array_api_ruleset) (function DType_bool () DType) (rewrite (Value_dtype (Value_bool b)) (DType_bool) :ruleset array_api_ruleset) (function Value_to_bool (Value) Boolean) @@ -274,19 +273,19 @@ (function DType_float32 () DType) (function IsDtypeKind_string (String) IsDtypeKind) (rewrite (isdtype (DType_float32) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) -(rewrite (isdtype %__expr_4101187900440652363 (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_-458881301673629543 (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) (function DType_object () DType) (rewrite (isdtype (DType_object) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int64) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) (function DType_int32 () DType) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_float32) (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) -(rewrite (isdtype %__expr_4101187900440652363 (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_-458881301673629543 (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_object) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int64) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_float32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) -(rewrite (isdtype %__expr_4101187900440652363 (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_-458881301673629543 (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_object) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int64) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) @@ -297,27 +296,27 @@ (function IsDtypeKind___or__ (IsDtypeKind IsDtypeKind) IsDtypeKind :cost 10) (rewrite (isdtype d (IsDtypeKind___or__ k1 k2)) (Boolean___or__ (isdtype d k1) (isdtype d k2)) :ruleset array_api_ruleset) (rewrite (IsDtypeKind___or__ k1 (IsDtypeKind_NULL)) k1 :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_4101187900440652363 %__expr_4101187900440652363) (TRUE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_float32)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_int32)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_int64)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_4101187900440652363 (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_float32) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_-458881301673629543 %__expr_-458881301673629543) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_float32)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_int32) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_float32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_int32)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_int64) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_float32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_int64)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_object) %__expr_4101187900440652363) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_float32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_int64)) (FALSE) :ruleset array_api_ruleset) @@ -434,119 +433,156 @@ (rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) (rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) +(extract (DType___eq__ %__expr_-458881301673629543 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (run-schedule (saturate (run array_api_ruleset))) -(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) 0) +(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_-4303590302573073911 (OptionalBool_none)) -(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(extract (Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) 0) +(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_8855076930384822898 (OptionalDevice_none)) -(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none)) (OptionalIntOrTuple_none))))) +(extract (isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) 0) +(let %__expr_4540085663845350143 (OptionalBool_none)) +(let %__expr_7530421370362517486 (OptionalDType_none)) +(let %__expr_-4082425080216461620 (OptionalDevice_none)) +(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) +(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) +(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) (run-schedule (saturate (run array_api_ruleset))) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1)) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) 0) +(let %__expr_7773480733401006441 (Int___init__ 1)) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) (run-schedule (saturate (run array_api_ruleset))) -(Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898)) (Int___init__ 3)) +(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) 0) +(Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (Int___init__ 3)) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_755392963466248965 (Int___init__ 2)) -(Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_755392963466248965) +(extract (Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (Int___init__ 3)) 0) +(let %__expr_-7750395295943998052 (Int___init__ 2)) +(Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) 0) +(TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_-990525301813695487 (OptionalDType_none)) -(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(extract (TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) +(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) +(TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_-4671014753414284575 (Int___init__ 0)) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(extract (TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) +(let %__expr_871583677900651896 (Int___init__ 0)) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (run-schedule (saturate (run array_api_ruleset))) -(Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) +(let %__expr_3713283541138310173 (Int___init__ 150)) +(Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(extract (Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_755392963466248965) +(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_-3636589015846229112 (Int___init__ 1)) -(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) +(extract (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) 0) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(extract (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) 0) +(TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) +(extract (TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) +(TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (run-schedule (saturate (run array_api_ruleset))) -(DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (DType_object)) +(extract (TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) 0) +(DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (DType_object)) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_755392963466248965) +(extract (DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (DType_object)) 0) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) (run-schedule (saturate (run array_api_ruleset))) -(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (IsDtypeKind_string "real floating")) +(extract (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) 0) +(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind_string "real floating")) (run-schedule (saturate (run array_api_ruleset))) -(Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) %__expr_-4671014753414284575) %__expr_755392963466248965) +(extract (isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind_string "real floating")) 0) +(Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) %__expr_871583677900651896) %__expr_-7750395295943998052) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) %__expr_871583677900651896) %__expr_-7750395295943998052) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_8800461850802719337 (Int___init__ 150)) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)) +(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)) (run-schedule (saturate (run array_api_ruleset))) -(TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)) 0) +(TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575) +(extract (TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) +(TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_-2515197891056208817 (Value_int (Int___init__ 0))) -(Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_-2515197891056208817))))) +(extract (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) 0) +(let %__expr_-2697243261424336135 (Value_int (Int___init__ 0))) +(Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_-2697243261424336135))))) (run-schedule (saturate (run array_api_ruleset))) -(Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) +(extract (Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_-2697243261424336135))))) 0) +(Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) (run-schedule (saturate (run array_api_ruleset))) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112)) +(extract (Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441)) (run-schedule (saturate (run array_api_ruleset))) -(Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112) %__expr_755392963466248965) +(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441)) 0) +(Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) %__expr_-7750395295943998052) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) %__expr_-7750395295943998052) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) +(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) +(TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575)) %__expr_-4671014753414284575) +(extract (TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) +(TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) %__expr_871583677900651896) (run-schedule (saturate (run array_api_ruleset))) -(NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) +(extract (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) %__expr_871583677900651896) 0) +(NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_-3193103899592592052 (Value_int (Int___init__ 1))) -(let %__expr_2090748637113203634 (Value_int (Int___init__ 2))) -(let %__expr_8968271287986727247 (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) -(let %__expr_-346696523742155791 (TupleNDArray___getitem__ %__expr_8968271287986727247 %__expr_-3636589015846229112)) -(let %__expr_888968520189910301 (NDArray_scalar %__expr_-2515197891056208817)) -(let %__expr_8768355756772896527 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) -(let %__expr_-294044290527551890 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) -(let %__expr_-8993155933335339035 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) +(extract (NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) 0) +(let %__expr_8044523218021318818 (Value_int %__expr_7773480733401006441)) +(let %__expr_-8852798553151442138 (Value_int (Int___init__ 2))) +(let %__expr_7039565710158856600 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) +(let %__expr_9189154200368037574 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) +(let %__expr_4948775059192706173 (NDArray_scalar %__expr_-2697243261424336135)) +(let %__expr_4396021293866471344 (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441)) +(let %__expr_4942768151788412671 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (sort MultiAxisIndexKeyItem) (sort Slice) (function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) (function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) -(let %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) -(let %__expr_-2454709417071966146 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) -(let %__expr_-277838035368029254 (OptionalInt_none)) -(let %__expr_-3045893352967658126 (FALSE)) +(let %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) +(let %__expr_6308258715381497904 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) +(sort IntOrTuple) +(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) +(function IntOrTuple_int (Int) IntOrTuple) +(let %__expr_-6383682287120045686 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896))) (sort MultiAxisIndexKey) (function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) (sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) (function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) (function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) -(let %__expr_469855974343835084 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-4671014753414284575) %__expr_1503884865394975690)))) -(let %__expr_5553578895160412584 (Float___init__ 1.0)) -(sort IntOrTuple) -(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) -(function IntOrTuple_int (Int) IntOrTuple) -(let %__expr_-4976863396675597554 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-4671014753414284575))) +(let %__expr_811525261394403186 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7750395295943998052) %__expr_4574319845551051569)))) +(let %__expr_-7672334842805227 (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) (function zeros (TupleInt OptionalDType OptionalDevice) NDArray) (function OptionalDType_some (DType) OptionalDType) @@ -556,55 +592,75 @@ (function mean (NDArray OptionalIntOrTuple Boolean) NDArray) (function IndexKey_ndarray (NDArray) IndexKey) (function NDArray___eq__ (NDArray NDArray) NDArray) -(let %__expr_6782829203688770802 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-4671014753414284575)) %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device %__expr_-2454709417071966146))) %__expr_469855974343835084 (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 %__expr_888968520189910301))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_-3193103899592592052)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_2090748637113203634)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126))) -(let %__expr_-8812288226011607914 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690)))) -(let %__expr_7373748767722314024 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690)))) -(function std (NDArray OptionalIntOrTuple) NDArray) +(let %__expr_4481754591798653144 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device %__expr_7039565710158856600))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 %__expr_4948775059192706173))) %__expr_-6383682287120045686 (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_8044523218021318818)))) %__expr_-6383682287120045686 (FALSE))) %__expr_811525261394403186 (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_-8852798553151442138)))) %__expr_-6383682287120045686 (FALSE)))) +(let %__expr_-907407888652541496 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569)))) +(let %__expr_5801044028809884995 (FALSE)) +(let %__expr_-6354454737791578173 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))) +(let %__expr_6624227409259184391 (Float___init__ 1.0)) (function OptionalInt_some (Int) OptionalInt) -(let %__expr_-4845818903945609769 (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-4671014753414284575))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575)) %__expr_-4976863396675597554)) -(let %__expr_-3351178202206693037 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-4671014753414284575))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575))) +(let %__expr_-7185048337851367263 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_871583677900651896))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-907407888652541496))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_7773480733401006441))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-6354454737791578173))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_-7750395295943998052))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_811525261394403186))))) (OptionalInt_some %__expr_871583677900651896))) (function NDArray___mul__ (NDArray NDArray) NDArray) (function ndarray-sqrt (NDArray) NDArray) -(TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126)) +(function std (NDArray OptionalIntOrTuple) NDArray) +(TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112) +(extract (TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) 0) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) (run-schedule (saturate (run array_api_ruleset))) -(let %__expr_6487687809380785968 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)) -(let %__expr_-3021532387541159817 (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_-277838035368029254)) -(let %__expr_5701152810033784528 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) -(let %__expr_-6187783262847860806 (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) -(let %__expr_-315001527591743086 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) -(let %__expr_-9037266425291862563 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126)) +(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) 0) +(let %__expr_-8392276577365791252 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(let %__expr_-9127408350583128512 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)) +(let %__expr_4873895522838289209 (OptionalInt_none)) +(let %__expr_1166113493069915661 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995) %__expr_7773480733401006441)) +(let %__expr_-2809124111957918512 (std %__expr_-7185048337851367263 %__expr_-6383682287120045686)) +(let %__expr_-1704542949091505620 (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(let %__expr_-2471947197794367259 (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_1166113493069915661 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_4873895522838289209)) +(let %__expr_5778560295018625428 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) (function NDArray___matmul__ (NDArray NDArray) NDArray) (function IndexKey_slice (Slice) IndexKey) -(TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_5701152810033784528)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) %__expr_-4671014753414284575) %__expr_-3636589015846229112))))))) (NDArray_T (NDArray___sub__ %__expr_6782829203688770802 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) %__expr_-3636589015846229112) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) %__expr_-3045893352967658126)) +(TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995)) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none))))) +(extract (TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995)) 0) +(TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none))))) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (Int___mul__ %__expr_-3636589015846229112 %__expr_6487687809380785968) %__expr_755392963466248965) +(extract (TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none))))) 0) +(Int___eq__ (Int___mul__ %__expr_7773480733401006441 %__expr_-9127408350583128512) %__expr_-7750395295943998052) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) +(extract (Int___eq__ (Int___mul__ %__expr_7773480733401006441 %__expr_-9127408350583128512) %__expr_-7750395295943998052) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (run-schedule (saturate (run array_api_ruleset))) -(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) 0) +(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) (run-schedule (saturate (run array_api_ruleset))) -(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(extract (Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) 0) +(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) (run-schedule (saturate (run array_api_ruleset))) -(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) %__expr_-4303590302573073911 (OptionalDevice_none)) (OptionalIntOrTuple_none))))) +(extract (isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) 0) +(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(Int___lt__ %__expr_5701152810033784528 %__expr_-3636589015846229112) +(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) +(Int___lt__ %__expr_-8392276577365791252 %__expr_7773480733401006441) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) +(extract (Int___lt__ %__expr_-8392276577365791252 %__expr_7773480733401006441) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) (run-schedule (saturate (run array_api_ruleset))) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1)) +(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) (run-schedule (saturate (run array_api_ruleset))) -(TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) +(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) 0) +(TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) (run-schedule (saturate (run array_api_ruleset))) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_-3636589015846229112) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_4101187900440652363) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1))) +(extract (TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) 0) +(let %__expr_-2157517542993472809 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_7773480733401006441) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441)) (run-schedule (saturate (run array_api_ruleset))) +(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_7773480733401006441) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441)) 0) (pop 1) (ruleset array_api_ruleset) (sort Int) @@ -1034,13 +1090,13 @@ (function Boolean___invert__ (Boolean) Boolean) (rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) (rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) -(ruleset ruleset_5064589968) +(ruleset ruleset_4998547360) (function NDArray___eq__ (NDArray NDArray) NDArray) -(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 1)) (NDArray_scalar (Value_int i))) (NDArray___eq__ x (NDArray_scalar (NDArray_index (unique_values x) (TupleInt_from_vec (vec-of i))))) :subsume :ruleset ruleset_5064589968) +(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 1)) (NDArray_scalar (Value_int i))) (NDArray___eq__ x (NDArray_scalar (NDArray_index (unique_values x) (TupleInt_from_vec (vec-of i))))) :subsume :ruleset ruleset_4998547360) (function count_values (NDArray NDArray) TupleValue :unextractable) -(rewrite (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (NDArray_vector (count_values x (unique_values x))) :subsume :ruleset ruleset_5064589968) -(rewrite (count_values x (NDArray_vector (TupleValue___add__ (TupleValue___init__ v) tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) (count_values x (NDArray_vector tv))) :subsume :ruleset ruleset_5064589968) -(rewrite (count_values x (NDArray_vector (TupleValue___init__ v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_5064589968) +(rewrite (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (NDArray_vector (count_values x (unique_values x))) :subsume :ruleset ruleset_4998547360) +(rewrite (count_values x (NDArray_vector (TupleValue___add__ (TupleValue___init__ v) tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) (count_values x (NDArray_vector tv))) :subsume :ruleset ruleset_4998547360) +(rewrite (count_values x (NDArray_vector (TupleValue___init__ v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_4998547360) (function std (NDArray OptionalIntOrTuple) NDArray) (sort IntOrTuple) (function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) @@ -1048,44 +1104,41 @@ (function ndarray-sqrt (NDArray) NDArray) (function mean (NDArray OptionalIntOrTuple Boolean) NDArray) (function square (NDArray) NDArray) -(rewrite (std x (OptionalIntOrTuple_some (IntOrTuple_int i))) (ndarray-sqrt (mean (square (NDArray___sub__ x (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE))) :subsume :ruleset ruleset_5064589968) -(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) :subsume :ruleset ruleset_5064589968) +(rewrite (std x (OptionalIntOrTuple_some (IntOrTuple_int i))) (ndarray-sqrt (mean (square (NDArray___sub__ x (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE))) :subsume :ruleset ruleset_4998547360) +(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) :subsume :ruleset ruleset_4998547360) (function expand_dims (NDArray Int) NDArray) -(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) i) :subsume :ruleset ruleset_5064589968) -(let %__expr_-4303590302573073911 (OptionalBool_none)) -(let %__expr_755392963466248965 (Int___init__ 2)) -(let %__expr_-990525301813695487 (OptionalDType_none)) -(let %__expr_8855076930384822898 (OptionalDevice_none)) +(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) i) :subsume :ruleset ruleset_4998547360) +(let %__expr_4540085663845350143 (OptionalBool_none)) +(let %__expr_7530421370362517486 (OptionalDType_none)) +(let %__expr_-4082425080216461620 (OptionalDevice_none)) (function NDArray_var (String) NDArray :cost 200) -(let %__expr_-8993155933335339035 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int %__expr_755392963466248965))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4303590302573073911) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) -(let %__expr_-294044290527551890 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray %__expr_-8993155933335339035 %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) (OptionalInt_none)))) -(let %__expr_-277838035368029254 (OptionalInt_none)) -(let %__expr_-3636589015846229112 (Int___init__ 1)) -(let %__expr_-2409712210977886203 (IndexKey_int (Int___init__ 0))) -(let %__expr_-4671014753414284575 (Int___init__ 0)) -(let %__expr_-3193103899592592052 (Value_int (Int___init__ 1))) -(let %__expr_2090748637113203634 (Value_int %__expr_755392963466248965)) -(let %__expr_8800461850802719337 (Int___init__ 150)) -(let %__expr_-2515197891056208817 (Value_int (Int___init__ 0))) -(let %__expr_8968271287986727247 (unique_inverse %__expr_-8993155933335339035)) -(let %__expr_-346696523742155791 (TupleNDArray___getitem__ %__expr_8968271287986727247 %__expr_-3636589015846229112)) -(let %__expr_888968520189910301 (NDArray_scalar %__expr_-2515197891056208817)) -(let %__expr_8768355756772896527 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898))) -(sort MultiAxisIndexKeyItem) -(sort Slice) -(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) -(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) -(let %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 %__expr_-277838035368029254 %__expr_-277838035368029254))) -(let %__expr_-2454709417071966146 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) -(let %__expr_-3045893352967658126 (FALSE)) +(let %__expr_9189154200368037574 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) +(let %__expr_7773480733401006441 (Int___init__ 1)) +(let %__expr_1906307530420321623 (IndexKey_int (Int___init__ 0))) +(let %__expr_871583677900651896 (Int___init__ 0)) +(let %__expr_4873895522838289209 (OptionalInt_none)) +(let %__expr_6308258715381497904 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray %__expr_9189154200368037574 %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) %__expr_4873895522838289209))) +(let %__expr_-7750395295943998052 (Int___init__ 2)) +(let %__expr_-8392276577365791252 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) +(let %__expr_-9127408350583128512 (TupleInt___getitem__ (NDArray_shape %__expr_6308258715381497904) %__expr_871583677900651896)) +(let %__expr_-8852798553151442138 (Value_int (Int___init__ 2))) +(let %__expr_8044523218021318818 (Value_int (Int___init__ 1))) +(let %__expr_3713283541138310173 (Int___init__ 150)) +(let %__expr_7039565710158856600 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) +(let %__expr_-2697243261424336135 (Value_int (Int___init__ 0))) (sort MultiAxisIndexKey) (function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) +(sort MultiAxisIndexKeyItem) (sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) (function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) (function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) -(let %__expr_469855974343835084 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-4671014753414284575) %__expr_1503884865394975690)))) -(let %__expr_5553578895160412584 (Float___init__ 1.0)) -(let %__expr_-4976863396675597554 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-4671014753414284575))) +(sort Slice) +(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) +(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) +(let %__expr_811525261394403186 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7750395295943998052) (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 %__expr_4873895522838289209 %__expr_4873895522838289209)))))) +(let %__expr_4396021293866471344 (TupleNDArray___getitem__ (unique_inverse %__expr_9189154200368037574) %__expr_7773480733401006441)) +(let %__expr_-7672334842805227 (unique_inverse %__expr_9189154200368037574)) +(let %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 %__expr_4873895522838289209 %__expr_4873895522838289209))) (function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) (function zeros (TupleInt OptionalDType OptionalDevice) NDArray) (function OptionalDType_some (DType) OptionalDType) @@ -1093,34 +1146,43 @@ (function OptionalDevice_some (Device) OptionalDevice) (function NDArray_device (NDArray) Device) (function IndexKey_ndarray (NDArray) IndexKey) -(let %__expr_6782829203688770802 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ %__expr_8968271287986727247 %__expr_-4671014753414284575)) %__expr_-4671014753414284575) (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-3636589015846229112))) (OptionalDType_some (NDArray_dtype %__expr_-2454709417071966146)) (OptionalDevice_some (NDArray_device %__expr_-2454709417071966146))) %__expr_469855974343835084 (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 %__expr_888968520189910301))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_-3193103899592592052)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690))) (mean (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-346696523742155791 (NDArray_scalar %__expr_2090748637113203634)))) %__expr_-4976863396675597554 %__expr_-3045893352967658126))) -(let %__expr_-8812288226011607914 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690)))) -(let %__expr_7373748767722314024 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_755392963466248965) %__expr_1503884865394975690)))) +(let %__expr_4481754591798653144 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ %__expr_-7672334842805227 %__expr_871583677900651896)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441))) (OptionalDType_some (NDArray_dtype %__expr_7039565710158856600)) (OptionalDevice_some (NDArray_device %__expr_7039565710158856600))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_-2697243261424336135)))) (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896)) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_8044523218021318818)))) (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896)) (FALSE))) %__expr_811525261394403186 (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_-8852798553151442138)))) (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896)) (FALSE)))) +(let %__expr_-907407888652541496 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569)))) +(let %__expr_-6354454737791578173 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))) +(let %__expr_4948775059192706173 (NDArray_scalar %__expr_-2697243261424336135)) +(let %__expr_6624227409259184391 (Float___init__ 1.0)) +(let %__expr_4942768151788412671 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(let %__expr_-6383682287120045686 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896))) +(let %__expr_5801044028809884995 (FALSE)) (function OptionalInt_some (Int) OptionalInt) -(let %__expr_-4845818903945609769 (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 %__expr_-2409712210977886203)))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575)) %__expr_-4976863396675597554)) -(let %__expr_-3351178202206693037 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 %__expr_-2409712210977886203)))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_469855974343835084))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_-3636589015846229112))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_-8812288226011607914))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-2454709417071966146 (IndexKey_ndarray (NDArray___eq__ %__expr_-8993155933335339035 (NDArray___getitem__ %__expr_-294044290527551890 (IndexKey_int %__expr_755392963466248965))))) (NDArray___getitem__ %__expr_6782829203688770802 %__expr_7373748767722314024))))) (OptionalInt_some %__expr_-4671014753414284575))) +(let %__expr_-7185048337851367263 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 %__expr_1906307530420321623)))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-907407888652541496))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_7773480733401006441))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-6354454737791578173))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_-7750395295943998052))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_811525261394403186))))) (OptionalInt_some %__expr_871583677900651896))) (function NDArray___mul__ (NDArray NDArray) NDArray) -(let %__expr_-3021532387541159817 (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_-277838035368029254)) -(let %__expr_5701152810033784528 (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575)) -(let %__expr_6487687809380785968 (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)) -(let %__expr_-6187783262847860806 (NDArray_dtype %__expr_-2454709417071966146)) -(let %__expr_-315001527591743086 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) -(let %__expr_-9037266425291862563 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ (TupleInt___getitem__ %__expr_8768355756772896527 %__expr_-4671014753414284575) (TupleInt___getitem__ (NDArray_shape %__expr_-294044290527551890) %__expr_-4671014753414284575)))))) %__expr_-990525301813695487 %__expr_-4303590302573073911 %__expr_8855076930384822898)) (NDArray___truediv__ %__expr_-3351178202206693037 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_-3045893352967658126)) +(let %__expr_1166113493069915661 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-8392276577365791252 %__expr_-9127408350583128512))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995) %__expr_7773480733401006441)) +(let %__expr_-2809124111957918512 (std %__expr_-7185048337851367263 %__expr_-6383682287120045686)) +(let %__expr_-1704542949091505620 (NDArray_dtype %__expr_7039565710158856600)) +(let %__expr_-2471947197794367259 (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_1166113493069915661 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_4873895522838289209)) +(let %__expr_5778560295018625428 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-8392276577365791252 %__expr_-9127408350583128512))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) (function NDArray___matmul__ (NDArray NDArray) NDArray) (function IndexKey_slice (Slice) IndexKey) -(let %__expr_-6769231790204798359 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_5701152810033784528)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ %__expr_6487687809380785968 %__expr_-3636589015846229112))))))) (NDArray_T (NDArray___sub__ %__expr_6782829203688770802 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) -(let %__expr_1884788319774111777 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_5701152810033784528)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_5553578895160412584 (Float_from_int (Int___sub__ %__expr_6487687809380785968 %__expr_-3636589015846229112))))))) (NDArray_T (NDArray___sub__ %__expr_6782829203688770802 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) %__expr_-3045893352967658126)) -(let %__expr_-5559224042962614522 (DType_int32)) -(let %__expr_-6676107104927312104 (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0))))) -(let %__expr_5888173354829332981 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-8993155933335339035) %__expr_-3636589015846229112) %__expr_-6187783262847860806) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_6782829203688770802)) -(let %__expr_2473377876388853102 (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) -(let %__expr_-3035386490804382424 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-9037266425291862563 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-3021532387541159817) %__expr_1503884865394975690)))) (NDArray___setitem__ %__expr_-4845818903945609769 (IndexKey_ndarray (NDArray___eq__ %__expr_-4845818903945609769 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_-315001527591743086 (IndexKey_slice %__expr_-3021532387541159817)))) -(simplify (saturate (seq (run array_api_ruleset) (run ruleset_5064589968))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-2454709417071966146 %__expr_5888173354829332981) (NDArray___matmul__ %__expr_-3035386490804382424 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_1884788319774111777 %__expr_755392963466248965)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_-6769231790204798359 (NDArray___mul__ (NDArray_scalar (Value_float (Float___init__ 0.0001))) (NDArray___getitem__ %__expr_-6769231790204798359 %__expr_-2409712210977886203))) %__expr_-5559224042962614522) (OptionalIntOrTuple_none))))) %__expr_-277838035368029254)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some %__expr_755392963466248965) %__expr_-277838035368029254))))))) +(let %__expr_2843694094811160801 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-9127408350583128512 %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995) %__expr_7773480733401006441)) +(let %__expr_-1190917143489565870 (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0))))) +(let %__expr_3952577209981612471 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-9127408350583128512 %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995)) +(let %__expr_-3429845910624471226 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) +(let %__expr_7346177422650294070 (DType_int32)) +(let %__expr_-7476478225777086989 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) +(let %__expr_-8563557304220193917 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144)) +(let %__expr_-5852474775256756550 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) +(simplify (saturate (seq (run array_api_ruleset) (run ruleset_4998547360))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_7039565710158856600 %__expr_-8563557304220193917) (NDArray___matmul__ %__expr_-5852474775256756550 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_3952577209981612471 %__expr_-7750395295943998052)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_2843694094811160801 (NDArray___mul__ %__expr_-3429845910624471226 (NDArray___getitem__ %__expr_2843694094811160801 %__expr_1906307530420321623))) %__expr_7346177422650294070) (OptionalIntOrTuple_none))))) %__expr_4873895522838289209)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some %__expr_-7750395295943998052) %__expr_4873895522838289209))))))) (pop 1) (sort Int) (function Int___init__ (i64) Int) -(let %__expr_-4671014753414284575 (Int___init__ 0)) -(let %__expr_8800461850802719337 (Int___init__ 150)) +(let %__expr_871583677900651896 (Int___init__ 0)) +(sort Value) +(function Value_int (Int) Value) +(let %__expr_8044523218021318818 (Value_int (Int___init__ 1))) +(let %__expr_7773480733401006441 (Int___init__ 1)) +(let %__expr_3188239554967046364 (Int___init__ 4)) +(let %__expr_3713283541138310173 (Int___init__ 150)) (sort NDArray) (sort IndexKey) (function NDArray___getitem__ (NDArray IndexKey) NDArray) @@ -1139,41 +1201,35 @@ (function assume_value_one_of (NDArray TupleValue) NDArray) (function DType_int64 () DType) (function TupleValue___add__ (TupleValue TupleValue) TupleValue) -(sort Value) (function TupleValue___init__ (Value) TupleValue) -(function Value_int (Int) Value) (function NDArray_scalar (Value) NDArray) -(let %__expr_-3473821386163038220 (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4))))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int %__expr_-4671014753414284575)))))) -(let %__expr_-3193103899592592052 (Value_int (Int___init__ 1))) -(let %__expr_2090748637113203634 (Value_int (Int___init__ 2))) -(let %__expr_-4688800770552726910 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337 (Int___init__ 4)))))) -(let %__expr_8326680938997940725 (NDArray___getitem__ %__expr_-4688800770552726910 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-3193103899592592052))))) -(sort OptionalIntOrTuple) -(sort IntOrTuple) -(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) -(function IntOrTuple_int (Int) IntOrTuple) -(let %__expr_-4976863396675597554 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-4671014753414284575))) -(let %__expr_-3636589015846229112 (Int___init__ 1)) -(let %__expr_-2298714861395141926 (Int___init__ 4)) -(let %__expr_-2515197891056208817 (Value_int %__expr_-4671014753414284575)) -(let %__expr_4101187900440652363 (DType_float64)) +(let %__expr_-8193044930268358572 (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173 %__expr_3188239554967046364)))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_8044523218021318818))))) +(let %__expr_-8852798553151442138 (Value_int (Int___init__ 2))) +(let %__expr_7894034366188933940 (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173 %__expr_3188239554967046364)))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-8852798553151442138))))) (sort MultiAxisIndexKeyItem) (sort Slice) (function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) (sort OptionalInt) (function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) (function OptionalInt_none () OptionalInt) -(let %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) -(let %__expr_5698022980052622903 (NDArray___getitem__ %__expr_-4688800770552726910 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_2090748637113203634))))) +(let %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) +(let %__expr_-8148879634944358227 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173 %__expr_3188239554967046364))))) +(let %__expr_-2697243261424336135 (Value_int %__expr_871583677900651896)) +(let %__expr_-6996286770058887658 (NDArray___getitem__ %__expr_-8148879634944358227 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-2697243261424336135))))) +(let %__expr_-6000263953540950166 (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2))))))) +(let %__expr_-7750395295943998052 (Int___init__ 2)) (sort MultiAxisIndexKey) (function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) (sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) (function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) (function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) -(let %__expr_-795645184301196957 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) %__expr_1503884865394975690)))) -(let %__expr_3323365144993085153 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-3636589015846229112) %__expr_1503884865394975690)))) -(let %__expr_7232724363220639299 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-4671014753414284575) %__expr_1503884865394975690)))) -(let %__expr_755392963466248965 (Int___init__ 2)) +(let %__expr_852382103983847011 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7750395295943998052) %__expr_4574319845551051569)))) +(let %__expr_-458881301673629543 (DType_float64)) +(sort OptionalIntOrTuple) +(sort IntOrTuple) +(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) +(function IntOrTuple_int (Int) IntOrTuple) +(let %__expr_-6383682287120045686 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896))) (function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) (sort OptionalDType) (sort OptionalDevice) @@ -1186,32 +1242,27 @@ (function sum (NDArray OptionalIntOrTuple) NDArray) (function TupleInt___getitem__ (TupleInt Int) Int) (function NDArray_shape (NDArray) TupleInt) -(let %__expr_-4084543222422775141 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) %__expr_-2298714861395141926)) (OptionalDType_some %__expr_4101187900440652363) (OptionalDevice_some (NDArray_device %__expr_-4688800770552726910))) %__expr_7232724363220639299 (NDArray___truediv__ (sum %__expr_-3473821386163038220 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-3473821386163038220) %__expr_-4671014753414284575))))) %__expr_3323365144993085153 (NDArray___truediv__ (sum %__expr_8326680938997940725 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_8326680938997940725) %__expr_-4671014753414284575))))) %__expr_-795645184301196957 (NDArray___truediv__ (sum %__expr_5698022980052622903 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_5698022980052622903) %__expr_-4671014753414284575)))))) +(let %__expr_1269092575758336087 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) %__expr_3188239554967046364)) (OptionalDType_some %__expr_-458881301673629543) (OptionalDevice_some (NDArray_device %__expr_-8148879634944358227))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569))) (NDArray___truediv__ (sum %__expr_-6996286770058887658 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-6996286770058887658) %__expr_871583677900651896))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569))) (NDArray___truediv__ (sum %__expr_-8193044930268358572 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-8193044930268358572) %__expr_871583677900651896))))) %__expr_852382103983847011 (NDArray___truediv__ (sum %__expr_7894034366188933940 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_7894034366188933940) %__expr_871583677900651896)))))) +(let %__expr_637321235555326848 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569)))) (sort TupleNDArray) (function concat (TupleNDArray OptionalInt) NDArray) (function TupleNDArray___add__ (TupleNDArray TupleNDArray) TupleNDArray) (function TupleNDArray___init__ (NDArray) TupleNDArray) (function NDArray___sub__ (NDArray NDArray) NDArray) (function OptionalInt_some (Int) OptionalInt) -(let %__expr_-5454083913435783283 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-3473821386163038220 (NDArray___getitem__ %__expr_-4084543222422775141 %__expr_7232724363220639299))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_8326680938997940725 (NDArray___getitem__ %__expr_-4084543222422775141 %__expr_3323365144993085153))) (TupleNDArray___init__ (NDArray___sub__ %__expr_5698022980052622903 (NDArray___getitem__ %__expr_-4084543222422775141 %__expr_-795645184301196957))))) (OptionalInt_some %__expr_-4671014753414284575))) -(let %__expr_4590898867152193220 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_2090748637113203634))) +(let %__expr_3786928033384172571 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-6996286770058887658 (NDArray___getitem__ %__expr_1269092575758336087 %__expr_637321235555326848))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-8193044930268358572 (NDArray___getitem__ %__expr_1269092575758336087 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))))) (TupleNDArray___init__ (NDArray___sub__ %__expr_7894034366188933940 (NDArray___getitem__ %__expr_1269092575758336087 %__expr_852382103983847011))))) (OptionalInt_some %__expr_871583677900651896))) (function OptionalIntOrTuple_none () OptionalIntOrTuple) -(let %__expr_2665112130965315478 (OptionalIntOrTuple_none)) -(let %__expr_-608551224937479330 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-3193103899592592052))) -(let %__expr_-5257952192217027907 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar (Value_int %__expr_-4671014753414284575)))) -(let %__expr_8511786851084399467 (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_8800461850802719337))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_-4671014753414284575)) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2))))))) +(let %__expr_8391745089977789030 (OptionalIntOrTuple_none)) +(let %__expr_6937128598082130908 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-8852798553151442138))) +(let %__expr_-7260687232290637626 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_8044523218021318818))) +(let %__expr_-167787087406110176 1) (function square (NDArray) NDArray) (function expand_dims (NDArray Int) NDArray) -(let %__expr_-1359368604848933744 (square (NDArray___sub__ %__expr_-5454083913435783283 (expand_dims (NDArray___truediv__ (sum %__expr_-5454083913435783283 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-5454083913435783283) %__expr_-4671014753414284575)))) %__expr_-4671014753414284575)))) -(let %__expr_-277838035368029254 (OptionalInt_none)) -(let %__expr_888968520189910301 (NDArray_scalar (Value_int %__expr_-4671014753414284575))) -(let %__expr_-3240316658536798486 1) +(let %__expr_8477700738181691910 (square (NDArray___sub__ %__expr_3786928033384172571 (expand_dims (NDArray___truediv__ (sum %__expr_3786928033384172571 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_3786928033384172571) %__expr_871583677900651896)))) %__expr_871583677900651896)))) +(let %__expr_8988307273465367651 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))) +(let %__expr_4857534354917498587 2) (function ndarray-sqrt (NDArray) NDArray) -(let %__expr_-2066297328423310743 (ndarray-sqrt (NDArray___truediv__ (sum %__expr_-1359368604848933744 %__expr_-4976863396675597554) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-1359368604848933744) %__expr_-4671014753414284575)))))) -(function Value_to_int (Value) Int) -(function NDArray_to_value (NDArray) Value) -(function astype (NDArray DType) NDArray) -(function NDArray___gt__ (NDArray NDArray) NDArray) +(let %__expr_3277279022187426799 (ndarray-sqrt (NDArray___truediv__ (sum %__expr_8477700738181691910 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_8477700738181691910) %__expr_871583677900651896)))))) (function TupleNDArray___getitem__ (TupleNDArray Int) NDArray) (sort Boolean) (function svd (NDArray Boolean) TupleNDArray) @@ -1220,290 +1271,299 @@ (function Value_float (Float) Value) (function Float_rational (Rational) Float :cost 2) (function FALSE () Boolean) +(let %__expr_-6992665957909292539 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 147))))) (NDArray___truediv__ %__expr_3786928033384172571 (NDArray___setitem__ %__expr_3277279022187426799 (IndexKey_ndarray (NDArray___eq__ %__expr_3277279022187426799 (NDArray_scalar %__expr_-2697243261424336135))) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_-167787087406110176))))))) (FALSE)) %__expr_7773480733401006441)) +(let %__expr_4948775059192706173 (NDArray_scalar %__expr_-2697243261424336135)) +(let %__expr_4873895522838289209 (OptionalInt_none)) +(function Value_to_int (Value) Int) +(function NDArray_to_value (NDArray) Value) +(function astype (NDArray DType) NDArray) +(function NDArray___gt__ (NDArray NDArray) NDArray) (function Float___init__ (f64) Float :cost 3) (function DType_int32 () DType) -(let %__expr_4399334460195737951 (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 147))))) (NDArray___truediv__ %__expr_-5454083913435783283 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486))))))) (FALSE)) %__expr_-3636589015846229112) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) %__expr_2665112130965315478)))) %__expr_-277838035368029254)) -(let %__expr_-6024165510290428751 150) +(let %__expr_5508951548654840646 (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_-6992665957909292539 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) %__expr_8391745089977789030)))) %__expr_4873895522838289209)) +(let %__expr_-2951635939159740441 150) (function NDArray_vector (TupleValue) NDArray) -(let %__expr_-4005821647467750755 (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-5257952192217027907 %__expr_2665112130965315478))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-608551224937479330 %__expr_2665112130965315478))) (TupleValue___init__ (NDArray_to_value (sum %__expr_4590898867152193220 %__expr_2665112130965315478)))))) %__expr_4101187900440652363) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-6024165510290428751 %__expr_-3240316658536798486)))))) -(let %__expr_-3931088046849885566 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 147))))) (NDArray___truediv__ %__expr_-5454083913435783283 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486))))))) (FALSE)) %__expr_-3636589015846229112)) -(let %__expr_-2315365344465410478 (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 147))))) (NDArray___truediv__ %__expr_-5454083913435783283 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486))))))) (FALSE))) -(let %__expr_-512142818984394169 (NDArray___setitem__ %__expr_-2066297328423310743 (IndexKey_ndarray (NDArray___eq__ %__expr_-2066297328423310743 %__expr_888968520189910301)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 %__expr_-3240316658536798486)))))) -(let %__expr_-3045893352967658126 (FALSE)) +(let %__expr_-3755709727314850279 (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-2697243261424336135)) %__expr_8391745089977789030))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-7260687232290637626 %__expr_8391745089977789030))) (TupleValue___init__ (NDArray_to_value (sum %__expr_6937128598082130908 %__expr_8391745089977789030)))))) %__expr_-458881301673629543) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-2951635939159740441 %__expr_-167787087406110176)))))) +(let %__expr_5561760508855517538 (NDArray___setitem__ %__expr_3277279022187426799 (IndexKey_ndarray (NDArray___eq__ %__expr_3277279022187426799 (NDArray_scalar %__expr_-2697243261424336135))) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_-167787087406110176)))))) +(let %__expr_8075847335762488630 (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 147))))) (NDArray___truediv__ %__expr_3786928033384172571 (NDArray___setitem__ %__expr_3277279022187426799 (IndexKey_ndarray (NDArray___eq__ %__expr_3277279022187426799 (NDArray_scalar %__expr_-2697243261424336135))) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_-167787087406110176))))))) (FALSE))) +(let %__expr_5801044028809884995 (FALSE)) (function NDArray___matmul__ (NDArray NDArray) NDArray) (function NDArray_T (NDArray) NDArray) (function IndexKey_slice (Slice) IndexKey) -(let %__expr_7487275443900416961 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_8800461850802719337)) %__expr_-4005821647467750755) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 2)))))) (NDArray_T (NDArray___sub__ %__expr_-4084543222422775141 (NDArray___matmul__ %__expr_-4005821647467750755 %__expr_-4084543222422775141))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-2315365344465410478 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_4399334460195737951) %__expr_1503884865394975690)))) %__expr_-512142818984394169)) (NDArray___getitem__ %__expr_-3931088046849885566 (IndexKey_slice %__expr_4399334460195737951)))) %__expr_-3045893352967658126) %__expr_-3636589015846229112)) -(let %__expr_6183362166771178405 2) -(let %__expr_-5395166694618893324 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_8800461850802719337)) %__expr_-4005821647467750755) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3240316658536798486 2)))))) (NDArray_T (NDArray___sub__ %__expr_-4084543222422775141 (NDArray___matmul__ %__expr_-4005821647467750755 %__expr_-4084543222422775141))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-2315365344465410478 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_4399334460195737951) %__expr_1503884865394975690)))) %__expr_-512142818984394169)) (NDArray___getitem__ %__expr_-3931088046849885566 (IndexKey_slice %__expr_4399334460195737951)))) %__expr_-3045893352967658126)) -(let %__expr_-5559224042962614522 (DType_int32)) -(let %__expr_2976592046696346842 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) -(let %__expr_-7009371984517665352 (NDArray___matmul__ %__expr_-4005821647467750755 %__expr_-4084543222422775141)) -(let %__expr_-6216319311343226647 (NDArray_var "y")) -(let %__expr_6240449845396421003 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-2315365344465410478 %__expr_755392963466248965) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_4399334460195737951) %__expr_1503884865394975690)))) %__expr_-512142818984394169)) (NDArray___getitem__ %__expr_-3931088046849885566 (IndexKey_slice %__expr_4399334460195737951)))) -(let %__expr_4033771585938608070 (NDArray_var "X")) +(let %__expr_-7611578709264289821 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_3713283541138310173)) %__expr_-3755709727314850279) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_4857534354917498587)))))) (NDArray_T (NDArray___sub__ %__expr_1269092575758336087 (NDArray___matmul__ %__expr_-3755709727314850279 %__expr_1269092575758336087))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_8075847335762488630 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_5508951548654840646) %__expr_4574319845551051569)))) %__expr_5561760508855517538)) (NDArray___getitem__ %__expr_-6992665957909292539 (IndexKey_slice %__expr_5508951548654840646)))) %__expr_5801044028809884995)) +(let %__expr_-3429845910624471226 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) +(let %__expr_7346177422650294070 (DType_int32)) +(let %__expr_-2626596927381929382 (NDArray_var "y")) +(let %__expr_-8462033748461398887 (NDArray___matmul__ %__expr_-3755709727314850279 %__expr_1269092575758336087)) +(let %__expr_3830746366128651238 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-2697243261424336135))) +(let %__expr_-1892903834471356524 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_8075847335762488630 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_5508951548654840646) %__expr_4574319845551051569)))) %__expr_5561760508855517538)) (NDArray___getitem__ %__expr_-6992665957909292539 (IndexKey_slice %__expr_5508951548654840646)))) +(let %__expr_5036000968216100388 (NDArray_var "X")) (sort Program) (function ndarray_function_two (NDArray NDArray NDArray) Program) (function IndexKey_int (Int) IndexKey) -(let %fn_program (ndarray_function_two (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-4688800770552726910 %__expr_-7009371984517665352) (NDArray___matmul__ %__expr_6240449845396421003 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-5395166694618893324 %__expr_755392963466248965)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_7487275443900416961 (NDArray___mul__ %__expr_2976592046696346842 (NDArray___getitem__ %__expr_7487275443900416961 (IndexKey_int %__expr_-4671014753414284575)))) %__expr_-5559224042962614522) %__expr_2665112130965315478)))) %__expr_-277838035368029254)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_1503884865394975690 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-277838035368029254 (OptionalInt_some %__expr_755392963466248965) %__expr_-277838035368029254)))))) %__expr_4033771585938608070 %__expr_-6216319311343226647)) -(ruleset ruleset_5064591888) +(let %fn_program (ndarray_function_two (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-8148879634944358227 %__expr_-8462033748461398887) (NDArray___matmul__ %__expr_-1892903834471356524 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-7611578709264289821 %__expr_-7750395295943998052)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ %__expr_-7611578709264289821 %__expr_7773480733401006441) (NDArray___mul__ %__expr_-3429845910624471226 (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-7611578709264289821 %__expr_7773480733401006441) (IndexKey_int %__expr_871583677900651896)))) %__expr_7346177422650294070) %__expr_8391745089977789030)))) %__expr_4873895522838289209)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some %__expr_-7750395295943998052) %__expr_4873895522838289209)))))) %__expr_5036000968216100388 %__expr_-2626596927381929382)) +(ruleset ruleset_4998548992) (function ndarray_program (NDArray) Program) (function Program___init__ (String bool) Program) -(rewrite (ndarray_program (NDArray_var s)) (Program___init__ s true) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray_var s)) (Program___init__ s true) :ruleset ruleset_4998548992) (function Program_statement (Program Program) Program :unextractable) (function Program___add__ (Program Program) Program) (function dtype_program (DType) Program) -(rewrite (ndarray_program (assume_dtype z dtype)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".dtype == " false)) (dtype_program dtype))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (assume_dtype z dtype)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".dtype == " false)) (dtype_program dtype))) :ruleset ruleset_4998548992) (function tuple_int_program (TupleInt) Program) -(rewrite (ndarray_program (assume_shape z ti)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".shape == " false)) (tuple_int_program ti))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (assume_isfinite z)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___init__ "assert np.all(np.isfinite(" false) (ndarray_program z)) (Program___init__ "))" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (assume_shape z ti)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".shape == " false)) (tuple_int_program ti))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (assume_isfinite z)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___init__ "assert np.all(np.isfinite(" false) (ndarray_program z)) (Program___init__ "))" false))) :ruleset ruleset_4998548992) (function tuple_value_program (TupleValue) Program) -(rewrite (ndarray_program (assume_value_one_of z tv)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert set(np.unique(" false) (ndarray_program z)) (Program___init__ ")) == set(" false)) (tuple_value_program tv)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (assume_value_one_of z tv)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert set(np.unique(" false) (ndarray_program z)) (Program___init__ ")) == set(" false)) (tuple_value_program tv)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (sort OptionalBool) (function reshape (NDArray TupleInt OptionalBool) NDArray) (function Program_assign (Program) Program) -(rewrite (ndarray_program (reshape y ti ob)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".reshape(" false)) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (astype y dtype)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".astype(" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (reshape y ti ob)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".reshape(" false)) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (astype y dtype)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".astype(" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function tuple_ndarray_program (TupleNDArray) Program) (function unique_counts (NDArray) TupleNDArray) -(rewrite (tuple_ndarray_program (unique_counts x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_counts=True)" false))) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program (unique_counts x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_counts=True)" false))) :ruleset ruleset_4998548992) (function unique_inverse (NDArray) TupleNDArray) -(rewrite (tuple_ndarray_program (unique_inverse x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_inverse=True)" false))) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program (unique_inverse x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_inverse=True)" false))) :ruleset ruleset_4998548992) (function int_program (Int) Program) -(rewrite (ndarray_program (TupleNDArray___getitem__ tnd i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program tnd) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (TupleNDArray___getitem__ tnd i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program tnd) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) (function value_program (Value) Program) -(rewrite (ndarray_program (NDArray_scalar v)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (value_program v)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray_scalar v)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (value_program v)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function OptionalDType_none () OptionalDType) -(rewrite (ndarray_program (zeros ti (OptionalDType_none) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (zeros ti (OptionalDType_some dtype) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ", dtype=" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (zeros ti (OptionalDType_none) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (zeros ti (OptionalDType_some dtype) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ", dtype=" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function unique_values (NDArray) NDArray) -(rewrite (ndarray_program (unique_values x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (unique_values x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function NDArray___add__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___add__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " + " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___sub__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " - " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___mul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " * " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___truediv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " / " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___add__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " + " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___sub__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " - " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___mul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " * " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___truediv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " / " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___lt__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___lt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " < " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___lt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " < " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___le__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___le__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " <= " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___gt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " > " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___le__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " <= " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___gt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " > " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___ge__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___ge__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >= " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___eq__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " == " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___matmul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " @ " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___ge__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >= " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___eq__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " == " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___matmul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " @ " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___mod__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___mod__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " % " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___mod__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " % " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___and__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___and__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " & " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___and__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " & " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___or__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___or__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " | " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___or__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " | " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___xor__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___xor__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ^ " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___xor__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ^ " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___lshift__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___lshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " << " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___lshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " << " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___rshift__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___rshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >> " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___rshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >> " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___floordiv__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___floordiv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " // " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___floordiv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " // " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function NDArray___pow__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___pow__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ** " false)) (ndarray_program y))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___pow__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ** " false)) (ndarray_program y))) :ruleset ruleset_4998548992) (function index_key_program (IndexKey) Program) -(rewrite (ndarray_program (NDArray___setitem__ x idx y)) (Program_statement (Program_assign (ndarray_program x)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program_assign (ndarray_program x)) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "] = " false)) (ndarray_program y))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray___getitem__ x idx)) (Program___add__ (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (square x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.square(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (expand_dims x i)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.expand_dims(" false) (ndarray_program x)) (Program___init__ ", " false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (NDArray___setitem__ x idx y)) (Program_statement (Program_assign (ndarray_program x)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program_assign (ndarray_program x)) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "] = " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___getitem__ x idx)) (Program___add__ (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (square x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.square(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (expand_dims x i)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.expand_dims(" false) (ndarray_program x)) (Program___init__ ", " false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function mean (NDArray OptionalIntOrTuple Boolean) NDArray) -(rewrite (ndarray_program (mean x %__expr_2665112130965315478 %__expr_-3045893352967658126)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (mean x %__expr_8391745089977789030 %__expr_5801044028809884995)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function int_or_tuple_program (IntOrTuple) Program) -(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) %__expr_-3045893352967658126)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) %__expr_5801044028809884995)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function TRUE () Boolean) -(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ", keepdims=True)" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (concat tnd %__expr_-277838035368029254)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (concat tnd (OptionalInt_some i))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ", axis=" false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray_vector tv)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (tuple_value_program tv)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ", keepdims=True)" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (concat tnd %__expr_4873895522838289209)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (concat tnd (OptionalInt_some i))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ", axis=" false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray_vector tv)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (tuple_value_program tv)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function std (NDArray OptionalIntOrTuple) NDArray) -(rewrite (ndarray_program (std x %__expr_2665112130965315478)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (std x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (tuple_ndarray_program (svd x (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (tuple_ndarray_program (svd x %__expr_-3045893352967658126)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ", full_matrices=False)" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (ndarray-sqrt x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sqrt(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (NDArray_T x)) (Program___add__ (ndarray_program x) (Program___init__ ".T" false)) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (sum x %__expr_2665112130965315478)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (sum x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) -(rewrite (tuple_int_program (NDArray_shape x)) (Program___add__ (ndarray_program x) (Program___init__ ".shape" false)) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (std x %__expr_8391745089977789030)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (std x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (tuple_ndarray_program (svd x (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (tuple_ndarray_program (svd x %__expr_5801044028809884995)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ", full_matrices=False)" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (ndarray-sqrt x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sqrt(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray_T x)) (Program___add__ (ndarray_program x) (Program___init__ ".T" false)) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (sum x %__expr_8391745089977789030)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (sum x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (tuple_int_program (NDArray_shape x)) (Program___add__ (ndarray_program x) (Program___init__ ".shape" false)) :ruleset ruleset_4998548992) (function ndarray-abs (NDArray) NDArray) -(rewrite (ndarray_program (ndarray-abs x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_5064591888) +(rewrite (ndarray_program (ndarray-abs x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) (function optional_int_or_tuple_program (OptionalIntOrTuple) Program) -(rewrite (optional_int_or_tuple_program (OptionalIntOrTuple_some it)) (int_or_tuple_program it) :ruleset ruleset_5064591888) -(rewrite (optional_int_or_tuple_program %__expr_2665112130965315478) (Program___init__ "None" false) :ruleset ruleset_5064591888) -(rewrite (int_or_tuple_program (IntOrTuple_int x)) (int_program x) :ruleset ruleset_5064591888) +(rewrite (optional_int_or_tuple_program (OptionalIntOrTuple_some it)) (int_or_tuple_program it) :ruleset ruleset_4998548992) +(rewrite (optional_int_or_tuple_program %__expr_8391745089977789030) (Program___init__ "None" false) :ruleset ruleset_4998548992) +(rewrite (int_or_tuple_program (IntOrTuple_int x)) (int_program x) :ruleset ruleset_4998548992) (function IntOrTuple_tuple (TupleInt) IntOrTuple) -(rewrite (int_or_tuple_program (IntOrTuple_tuple t)) (tuple_int_program t) :ruleset ruleset_5064591888) +(rewrite (int_or_tuple_program (IntOrTuple_tuple t)) (tuple_int_program t) :ruleset ruleset_4998548992) (function IndexKey_ELLIPSIS () IndexKey) -(rewrite (index_key_program (IndexKey_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_5064591888) -(rewrite (index_key_program (IndexKey_int i)) (int_program i) :ruleset ruleset_5064591888) +(rewrite (index_key_program (IndexKey_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_4998548992) +(rewrite (index_key_program (IndexKey_int i)) (int_program i) :ruleset ruleset_4998548992) (function slice_program (Slice) Program) -(rewrite (index_key_program (IndexKey_slice s)) (slice_program s) :ruleset ruleset_5064591888) +(rewrite (index_key_program (IndexKey_slice s)) (slice_program s) :ruleset ruleset_4998548992) (function multi_axis_index_key_program (MultiAxisIndexKey) Program) -(rewrite (index_key_program (IndexKey_multi_axis key)) (multi_axis_index_key_program key) :ruleset ruleset_5064591888) -(rewrite (index_key_program (IndexKey_ndarray a)) (ndarray_program a) :ruleset ruleset_5064591888) +(rewrite (index_key_program (IndexKey_multi_axis key)) (multi_axis_index_key_program key) :ruleset ruleset_4998548992) +(rewrite (index_key_program (IndexKey_ndarray a)) (ndarray_program a) :ruleset ruleset_4998548992) (sort UnstableFn_MultiAxisIndexKeyItem_Int (UnstableFn (Int) MultiAxisIndexKeyItem)) (function Int___add__ (Int Int) Int) -(function idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ (UnstableFn_MultiAxisIndexKeyItem_Int Int) MultiAxisIndexKeyItem) -(rewrite (idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_-3636589015846229112)) :ruleset ruleset_5064591888) +(function idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ (UnstableFn_MultiAxisIndexKeyItem_Int Int) MultiAxisIndexKeyItem) +(rewrite (idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_7773480733401006441)) :ruleset ruleset_4998548992) (function MultiAxisIndexKey___init__ (Int UnstableFn_MultiAxisIndexKeyItem_Int) MultiAxisIndexKey) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ %__expr_-4671014753414284575 idx_fn)) (Program___init__ "" false) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ %__expr_871583677900651896 idx_fn)) (Program___init__ "" false) :ruleset ruleset_4998548992) (function multi_axis_index_key_item_program (MultiAxisIndexKeyItem) Program) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (unstable-app idx_fn %__expr_-4671014753414284575)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ (- k %__expr_-3240316658536798486)) (unstable-fn "idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_5064591888) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_5064591888) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ "," false)) :when ((= (vec-length vec) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-remove vec 0)))) :when ((> (vec-length vec) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_int i)) (int_program i) :ruleset ruleset_5064591888) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_slice s)) (slice_program s) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (unstable-app idx_fn %__expr_871583677900651896)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ (- k %__expr_-167787087406110176)) (unstable-fn "idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ "," false)) :when ((= (vec-length vec) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-remove vec 0)))) :when ((> (vec-length vec) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_int i)) (int_program i) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_slice s)) (slice_program s) :ruleset ruleset_4998548992) (function MultiAxisIndexKeyItem_ELLIPSIS () MultiAxisIndexKeyItem) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_4998548992) (function MultiAxisIndexKeyItem_NONE () MultiAxisIndexKeyItem) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_NONE)) (Program___init__ "None" false) :ruleset ruleset_5064591888) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_NONE)) (Program___init__ "None" false) :ruleset ruleset_4998548992) (function optional_int_slice_program (OptionalInt) Program) -(rewrite (slice_program (Slice___init__ start stop %__expr_-277838035368029254)) (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) :ruleset ruleset_5064591888) -(rewrite (slice_program (Slice___init__ start stop (OptionalInt_some i))) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) (Program___init__ ":" false)) (int_program i)) :ruleset ruleset_5064591888) -(rewrite (optional_int_slice_program %__expr_-277838035368029254) (Program___init__ "" false) :ruleset ruleset_5064591888) -(rewrite (optional_int_slice_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_5064591888) +(rewrite (slice_program (Slice___init__ start stop %__expr_4873895522838289209)) (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) :ruleset ruleset_4998548992) +(rewrite (slice_program (Slice___init__ start stop (OptionalInt_some i))) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) (Program___init__ ":" false)) (int_program i)) :ruleset ruleset_4998548992) +(rewrite (optional_int_slice_program %__expr_4873895522838289209) (Program___init__ "" false) :ruleset ruleset_4998548992) +(rewrite (optional_int_slice_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_4998548992) (function optional_int_program (OptionalInt) Program) -(rewrite (optional_int_program %__expr_-277838035368029254) (Program___init__ "None" false) :ruleset ruleset_5064591888) -(rewrite (optional_int_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_5064591888) +(rewrite (optional_int_program %__expr_4873895522838289209) (Program___init__ "None" false) :ruleset ruleset_4998548992) +(rewrite (optional_int_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_4998548992) (function optional_dtype_program (OptionalDType) Program) -(rewrite (optional_dtype_program (OptionalDType_none)) (Program___init__ "None" false) :ruleset ruleset_5064591888) -(rewrite (optional_dtype_program (OptionalDType_some dtype)) (dtype_program dtype) :ruleset ruleset_5064591888) +(rewrite (optional_dtype_program (OptionalDType_none)) (Program___init__ "None" false) :ruleset ruleset_4998548992) +(rewrite (optional_dtype_program (OptionalDType_some dtype)) (dtype_program dtype) :ruleset ruleset_4998548992) (function tuple_ndarray_program_inner (TupleNDArray) Program) -(rewrite (tuple_ndarray_program r) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_ndarray_program_inner r)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) -(rewrite (tuple_ndarray_program_inner (TupleNDArray___init__ x)) (Program___add__ (ndarray_program x) (Program___init__ "," false)) :ruleset ruleset_5064591888) -(rewrite (tuple_ndarray_program_inner (TupleNDArray___add__ l r)) (Program___add__ (Program___add__ (tuple_ndarray_program_inner l) (Program___init__ " " false)) (tuple_ndarray_program_inner r)) :ruleset ruleset_5064591888) +(rewrite (tuple_ndarray_program r) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_ndarray_program_inner r)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (tuple_ndarray_program_inner (TupleNDArray___init__ x)) (Program___add__ (ndarray_program x) (Program___init__ "," false)) :ruleset ruleset_4998548992) +(rewrite (tuple_ndarray_program_inner (TupleNDArray___add__ l r)) (Program___add__ (Program___add__ (tuple_ndarray_program_inner l) (Program___init__ " " false)) (tuple_ndarray_program_inner r)) :ruleset ruleset_4998548992) (function TupleNDArray_length (TupleNDArray) Int) -(rewrite (int_program (TupleNDArray_length l)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_ndarray_program l)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) -(rewrite (ndarray_program (TupleNDArray___getitem__ l i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program l) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (TupleNDArray_length l)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_ndarray_program l)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (TupleNDArray___getitem__ l i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program l) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) (function tuple_value_program_inner (TupleValue) Program) -(rewrite (tuple_value_program tv1) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_value_program_inner tv1)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) -(rewrite (tuple_value_program_inner (TupleValue___add__ tv1 tv2)) (Program___add__ (Program___add__ (tuple_value_program_inner tv1) (Program___init__ " " false)) (tuple_value_program_inner tv2)) :ruleset ruleset_5064591888) -(rewrite (tuple_value_program_inner (TupleValue___init__ v)) (Program___add__ (value_program v) (Program___init__ "," false)) :ruleset ruleset_5064591888) -(rewrite (value_program (Value_int i)) (int_program i) :ruleset ruleset_5064591888) +(rewrite (tuple_value_program tv1) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_value_program_inner tv1)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (tuple_value_program_inner (TupleValue___add__ tv1 tv2)) (Program___add__ (Program___add__ (tuple_value_program_inner tv1) (Program___init__ " " false)) (tuple_value_program_inner tv2)) :ruleset ruleset_4998548992) +(rewrite (tuple_value_program_inner (TupleValue___init__ v)) (Program___add__ (value_program v) (Program___init__ "," false)) :ruleset ruleset_4998548992) +(rewrite (value_program (Value_int i)) (int_program i) :ruleset ruleset_4998548992) (function Value_bool (Boolean) Value) (function bool_program (Boolean) Program) -(rewrite (value_program (Value_bool b)) (bool_program b) :ruleset ruleset_5064591888) +(rewrite (value_program (Value_bool b)) (bool_program b) :ruleset ruleset_4998548992) (function float_program (Float) Program) -(rewrite (value_program (Value_float f)) (float_program f) :ruleset ruleset_5064591888) -(rewrite (value_program (NDArray_to_value x)) (ndarray_program x) :ruleset ruleset_5064591888) +(rewrite (value_program (Value_float f)) (float_program f) :ruleset ruleset_4998548992) +(rewrite (value_program (NDArray_to_value x)) (ndarray_program x) :ruleset ruleset_4998548992) (function Value___lt__ (Value Value) Value) -(rewrite (value_program (Value___lt__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " < " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (value_program (Value___lt__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " < " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Value___truediv__ (Value Value) Value) -(rewrite (value_program (Value___truediv__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " / " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (value_program (Value___truediv__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " / " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Value_to_bool (Value) Boolean) -(rewrite (bool_program (Value_to_bool v1)) (value_program v1) :ruleset ruleset_5064591888) -(rewrite (int_program (Value_to_int v1)) (value_program v1) :ruleset ruleset_5064591888) -(rewrite (float_program (Float___init__ f64_)) (Program___init__ (to-string f64_) false) :ruleset ruleset_5064591888) +(rewrite (bool_program (Value_to_bool v1)) (value_program v1) :ruleset ruleset_4998548992) +(rewrite (int_program (Value_to_int v1)) (value_program v1) :ruleset ruleset_4998548992) +(rewrite (float_program (Float___init__ f64_)) (Program___init__ (to-string f64_) false) :ruleset ruleset_4998548992) (function Float_abs (Float) Float) -(rewrite (float_program (Float_abs f)) (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (float_program f)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (float_program (Float_abs f)) (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (float_program f)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Float_from_int (Int) Float) -(rewrite (float_program (Float_from_int i)) (int_program i) :ruleset ruleset_5064591888) +(rewrite (float_program (Float_from_int i)) (int_program i) :ruleset ruleset_4998548992) (function Float___add__ (Float Float) Float) -(rewrite (float_program (Float___add__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " + " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (float_program (Float___add__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " + " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Float___sub__ (Float Float) Float) -(rewrite (float_program (Float___sub__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " - " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (float_program (Float___sub__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " - " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Float___mul__ (Float Float) Float) -(rewrite (float_program (Float___mul__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " * " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (float_program (Float___mul__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " * " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Float___truediv__ (Float Float) Float) -(rewrite (float_program (Float___truediv__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " / " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) -(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ " / " false)) (Program___init__ (to-string (denom r)) false)) (Program___init__ ")" false)) :when ((!= (denom r) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) -(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ ")" false)) :when ((= (denom r) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) -(rewrite (dtype_program %__expr_4101187900440652363) (Program___init__ "np.dtype(np.float64)" false) :ruleset ruleset_5064591888) +(rewrite (float_program (Float___truediv__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " / " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ " / " false)) (Program___init__ (to-string (denom r)) false)) (Program___init__ ")" false)) :when ((!= (denom r) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) +(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ ")" false)) :when ((= (denom r) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) +(rewrite (dtype_program %__expr_-458881301673629543) (Program___init__ "np.dtype(np.float64)" false) :ruleset ruleset_4998548992) (function DType_float32 () DType) -(rewrite (dtype_program (DType_float32)) (Program___init__ "np.dtype(np.float32)" false) :ruleset ruleset_5064591888) -(rewrite (dtype_program (DType_int64)) (Program___init__ "np.dtype(np.int64)" false) :ruleset ruleset_5064591888) -(rewrite (dtype_program %__expr_-5559224042962614522) (Program___init__ "np.dtype(np.int32)" false) :ruleset ruleset_5064591888) +(rewrite (dtype_program (DType_float32)) (Program___init__ "np.dtype(np.float32)" false) :ruleset ruleset_4998548992) +(rewrite (dtype_program (DType_int64)) (Program___init__ "np.dtype(np.int64)" false) :ruleset ruleset_4998548992) +(rewrite (dtype_program %__expr_7346177422650294070) (Program___init__ "np.dtype(np.int32)" false) :ruleset ruleset_4998548992) (function DType_bool () DType) -(rewrite (dtype_program (DType_bool)) (Program___init__ "np.dtype(np.bool)" false) :ruleset ruleset_5064591888) +(rewrite (dtype_program (DType_bool)) (Program___init__ "np.dtype(np.bool)" false) :ruleset ruleset_4998548992) (function DType_object () DType) -(rewrite (dtype_program (DType_object)) (Program___init__ "np.dtype(np.object_)" false) :ruleset ruleset_5064591888) +(rewrite (dtype_program (DType_object)) (Program___init__ "np.dtype(np.object_)" false) :ruleset ruleset_4998548992) (function Program_function_two (Program Program Program String) Program) +(function Program_compile (Program i64) Unit :default ()) +(rule ((= f (ndarray_function_two res l r))) + ((union f (Program_function_two (ndarray_program res) (ndarray_program l) (ndarray_program r) "__fn")) + (Program_compile f 0)) + :ruleset ruleset_4998548992 ) (sort UnstableFn_Int_Int (UnstableFn (Int) Int)) -(function idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ (UnstableFn_Int_Int Int) Int) -(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_-3636589015846229112)) :ruleset ruleset_5064591888) -(rewrite (int_program (TupleInt___getitem__ ti i)) (Program___add__ (Program___add__ (Program___add__ (tuple_int_program ti) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_5064591888) +(function idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ (UnstableFn_Int_Int Int) Int) +(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_7773480733401006441)) :ruleset ruleset_4998548992) +(rewrite (int_program (TupleInt___getitem__ ti i)) (Program___add__ (Program___add__ (Program___add__ (tuple_int_program ti) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) (function TupleInt_length (TupleInt) Int) -(rewrite (int_program (TupleInt_length ti)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_int_program ti)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (TupleInt_length ti)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_int_program ti)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function tuple_int_program_inner (TupleInt) Program) -(rewrite (tuple_int_program ti) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_int_program_inner ti)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program ti) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_int_program_inner ti)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt :unextractable) -(rewrite (tuple_int_program_inner (TupleInt___init__ %__expr_-4671014753414284575 idx_fn)) (Program___init__ "" false) :ruleset ruleset_5064591888) -(rewrite (tuple_int_program_inner (TupleInt___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (int_program (unstable-app idx_fn %__expr_-4671014753414284575)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt___init__ (Int___init__ (- k %__expr_-3240316658536798486)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_-3636589015846229112__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_5064591888) -(rewrite (tuple_int_program_inner (TupleInt_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_5064591888) -(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt_from_vec (vec-remove vec_int 0)))) :when ((> (vec-length vec_int) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) -(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ "," false)) :when ((= (vec-length vec_int) %__expr_-3240316658536798486)) :ruleset ruleset_5064591888) -(rewrite (int_program (Int___init__ i64_)) (Program___init__ (to-string i64_) false) :ruleset ruleset_5064591888) +(rewrite (tuple_int_program_inner (TupleInt___init__ %__expr_871583677900651896 idx_fn)) (Program___init__ "" false) :ruleset ruleset_4998548992) +(rewrite (tuple_int_program_inner (TupleInt___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (int_program (unstable-app idx_fn %__expr_871583677900651896)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt___init__ (Int___init__ (- k %__expr_-167787087406110176)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_4998548992) +(rewrite (tuple_int_program_inner (TupleInt_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_4998548992) +(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt_from_vec (vec-remove vec_int 0)))) :when ((> (vec-length vec_int) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) +(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ "," false)) :when ((= (vec-length vec_int) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) +(rewrite (int_program (Int___init__ i64_)) (Program___init__ (to-string i64_) false) :ruleset ruleset_4998548992) (function Int___invert__ (Int) Int) -(rewrite (int_program (Int___invert__ i)) (Program___add__ (Program___init__ "~" false) (int_program i)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___invert__ i)) (Program___add__ (Program___init__ "~" false) (int_program i)) :ruleset ruleset_4998548992) (function Int___lt__ (Int Int) Boolean) -(rewrite (bool_program (Int___lt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " < " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (bool_program (Int___lt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " < " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___le__ (Int Int) Boolean) -(rewrite (bool_program (Int___le__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " <= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (bool_program (Int___le__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " <= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___gt__ (Int Int) Boolean) -(rewrite (bool_program (Int___gt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " > " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (bool_program (Int___gt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " > " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___ge__ (Int Int) Boolean) -(rewrite (bool_program (Int___ge__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (bool_program (Int___ge__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___eq__ (Int Int) Boolean) -(rewrite (bool_program (Int___eq__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " == " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) -(rewrite (int_program (Int___add__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " + " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (bool_program (Int___eq__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " == " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (int_program (Int___add__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " + " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___sub__ (Int Int) Int) -(rewrite (int_program (Int___sub__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " - " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___sub__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " - " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___mul__ (Int Int) Int) -(rewrite (int_program (Int___mul__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " * " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___mul__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " * " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___truediv__ (Int Int) Int) -(rewrite (int_program (Int___truediv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " / " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___truediv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " / " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___mod__ (Int Int) Int) -(rewrite (int_program (Int___mod__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " % " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___mod__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " % " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___pow__ (Int Int) Int) -(rewrite (int_program (Int___pow__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ** " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___pow__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ** " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___and__ (Int Int) Int) -(rewrite (int_program (Int___and__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " & " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___and__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " & " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___or__ (Int Int) Int) -(rewrite (int_program (Int___or__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " | " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___or__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " | " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___xor__ (Int Int) Int) -(rewrite (int_program (Int___xor__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ^ " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___xor__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ^ " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___lshift__ (Int Int) Int) -(rewrite (int_program (Int___lshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " << " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___lshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " << " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___rshift__ (Int Int) Int) -(rewrite (int_program (Int___rshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >> " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) +(rewrite (int_program (Int___rshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >> " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) (function Int___floordiv__ (Int Int) Int) -(rewrite (int_program (Int___floordiv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " // " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_5064591888) -(rewrite (bool_program (TRUE)) (Program___init__ "True" false) :ruleset ruleset_5064591888) -(rewrite (bool_program %__expr_-3045893352967658126) (Program___init__ "False" false) :ruleset ruleset_5064591888) -(ruleset ruleset_5063997584) +(rewrite (int_program (Int___floordiv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " // " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (bool_program (TRUE)) (Program___init__ "True" false) :ruleset ruleset_4998548992) +(rewrite (bool_program %__expr_5801044028809884995) (Program___init__ "False" false) :ruleset ruleset_4998548992) +(ruleset ruleset_4997969680) (function Program_expr (Program) String) (function Program_statements (Program) String) (function Program_next_sym (Program) i64) (function Program_is_identifer (Program) bool) -(function Program_compile (Program i64) Unit :default ()) (rule ((= p (Program___init__ s b)) (Program_compile p i)) ((set (Program_expr p) s) (set (Program_statements p) "") (set (Program_next_sym p) i) (set (Program_is_identifer p) b)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (function Program_expr_to_statement (Program) Program) -(rewrite (Program_statement p1 p2) (Program___add__ p1 (Program_expr_to_statement p2)) :ruleset ruleset_5063997584) +(rewrite (Program_statement p1 p2) (Program___add__ p1 (Program_expr_to_statement p2)) :ruleset ruleset_4997969680) (function Program_parent (Program) Program :unextractable :merge old) (rule ((= p (Program_expr_to_statement p1)) (Program_compile p i)) ((set (Program_parent p1) p) (set (Program_is_identifer p) false)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_expr_to_statement p1)) (Program_compile p i) (= (Program_parent p1) p)) ((Program_compile p1 i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_expr_to_statement p1)) (Program_compile p i) (!= (Program_parent p1) p) @@ -1512,7 +1572,7 @@ ")) (set (Program_next_sym p) i) (set (Program_expr p) "")) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_expr_to_statement p1)) (= (Program_parent p1) p) (= s1 (Program_expr p1)) @@ -1522,54 +1582,54 @@ ")) (set (Program_next_sym p) i) (set (Program_expr p) "")) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (= (Program_expr p) (Program_expr p1)) (= b (Program_is_identifer p1))) ((set (Program_is_identifer p) b)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (= (Program_expr p) (Program_expr p2)) (= b (Program_is_identifer p2))) ((set (Program_is_identifer p) b)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (!= (Program_expr p) (Program_expr p1)) (!= (Program_expr p) (Program_expr p2))) ((set (Program_is_identifer p) false)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i)) ((set (Program_parent p1) p)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (= (Program_parent p1) p)) ((Program_compile p1 i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (Program_next_sym p1)) ((set (Program_parent p2) p)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (!= (Program_parent p1) p) (= (Program_parent p2) p)) ((Program_compile p2 i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i2) (= (Program_parent p1) p) (= i (Program_next_sym p1)) (= (Program_parent p2) p)) ((Program_compile p2 i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (= s1 (Program_expr p1)) (= s2 (Program_expr p2))) ((set (Program_expr p) (+ s1 s2))) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (= (Program_parent p1) p) (= (Program_parent p2) p) @@ -1578,14 +1638,14 @@ (= i (Program_next_sym p2))) ((set (Program_statements p) (+ s1 s2)) (set (Program_next_sym p) i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (!= (Program_parent p1) p) (!= (Program_parent p2) p)) ((set (Program_statements p) "") (set (Program_next_sym p) i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (= (Program_parent p1) p) (!= (Program_parent p2) p) @@ -1593,7 +1653,7 @@ (= i (Program_next_sym p1))) ((set (Program_statements p) s1) (set (Program_next_sym p) i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program___add__ p1 p2)) (= (Program_parent p2) p) (!= (Program_parent p1) p) @@ -1601,17 +1661,17 @@ (= i (Program_next_sym p2))) ((set (Program_statements p) s2) (set (Program_next_sym p) i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_assign p1)) (Program_compile p i)) ((set (Program_parent p1) p) (set (Program_is_identifer p) true)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_assign p1)) (Program_compile p i) (= (Program_parent p1) p)) ((Program_compile p1 i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_assign p1)) (= (Program_parent p1) p) (= s1 (Program_statements p1)) @@ -1621,8 +1681,8 @@ ((set (Program_statements p) (+ s1 (+ "_" (to-string i)) " = " s2 " ")) (set (Program_expr p) (+ "_" (to-string i))) - (set (Program_next_sym p) (+ i %__expr_-3240316658536798486))) - :ruleset ruleset_5063997584 ) + (set (Program_next_sym p) (+ i %__expr_-167787087406110176))) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_assign p1)) (!= (Program_parent p1) p) (Program_compile p i) @@ -1631,8 +1691,8 @@ ((set (Program_statements p) (+ (+ "_" (to-string i)) " = " s2 " ")) (set (Program_expr p) (+ "_" (to-string i))) - (set (Program_next_sym p) (+ i %__expr_-3240316658536798486))) - :ruleset ruleset_5063997584 ) + (set (Program_next_sym p) (+ i %__expr_-167787087406110176))) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_assign p1)) (= (Program_parent p1) p) (= s1 (Program_statements p1)) @@ -1642,7 +1702,7 @@ ((set (Program_statements p) s1) (set (Program_expr p) s2) (set (Program_next_sym p) i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_assign p1)) (!= (Program_parent p1) p) (Program_compile p i) @@ -1651,7 +1711,7 @@ ((set (Program_statements p) "") (set (Program_expr p) s2) (set (Program_next_sym p) i)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_function_two p1 p2 p3 s1)) (Program_compile p i)) ((set (Program_parent p2) p) @@ -1661,7 +1721,7 @@ (Program_compile p3 i) (Program_compile p1 i) (set (Program_is_identifer p) true)) - :ruleset ruleset_5063997584 ) + :ruleset ruleset_4997969680 ) (rule ((= p (Program_function_two p1 p2 p3 s1)) (Program_compile p i) (= s2 (Program_expr p1)) @@ -1675,10 +1735,6 @@ ")) (set (Program_next_sym p) i) (set (Program_expr p) s1)) - :ruleset ruleset_5063997584 ) -(rule ((= f (ndarray_function_two res l r))) - ((union f (Program_function_two (ndarray_program res) (ndarray_program l) (ndarray_program r) "__fn")) - (Program_compile f 0)) - :ruleset ruleset_5064591888 ) -(run-schedule (seq (saturate (run ruleset_5064591888)) (saturate (run ruleset_5063997584)))) + :ruleset ruleset_4997969680 ) +(run-schedule (seq (saturate (run ruleset_4998548992)) (saturate (run ruleset_4997969680)))) (extract (Program_statements %fn_program) 0) From d684867551503130f6a25ec6ea8f1efa8080f15f Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 17 Oct 2024 17:13:11 -0400 Subject: [PATCH 03/11] Only run long running benchmarks https://github.com/oxc-project/backlog/issues/89 --- benches/example_benchmarks.rs | 25 ++++++++++++++++--------- 1 file changed, 16 insertions(+), 9 deletions(-) diff --git a/benches/example_benchmarks.rs b/benches/example_benchmarks.rs index e733986a..548942c7 100644 --- a/benches/example_benchmarks.rs +++ b/benches/example_benchmarks.rs @@ -1,6 +1,19 @@ use codspeed_criterion_compat::{criterion_group, criterion_main, Criterion}; use egglog::EGraph; +// Only benchmark longer running examples, +// because many of the short ones have too much variance due to the allocator being non deterministic. +// https://github.com/oxc-project/backlog/issues/89 + +const BENCHMARKS: &[&str] = &[ + "eggcc-extraction", + "math-microbenchmark", + "herbie", + "typeinfer", + "lambda", + "python_array_optimize", +]; + fn run_example(filename: &str, program: &str) { EGraph::default() .parse_and_run_program(Some(filename.to_owned()), program) @@ -8,16 +21,10 @@ fn run_example(filename: &str, program: &str) { } pub fn criterion_benchmark(c: &mut Criterion) { - for entry in glob::glob("tests/**/*.egg").unwrap() { - let path = entry.unwrap().clone(); - let path_string = path.to_string_lossy().to_string(); - if path_string.contains("fail-typecheck") { - continue; - } - let name = path.file_stem().unwrap().to_string_lossy().to_string(); - let filename = path.to_string_lossy().to_string(); + for name in BENCHMARKS { + let filename = format!("tests/{}.egg", name); let program = std::fs::read_to_string(&filename).unwrap(); - c.bench_function(&name, |b| b.iter(|| run_example(&filename, &program))); + c.bench_function(name, |b| b.iter(|| run_example(&filename, &program))); } } From e25b2f3f2b0f9eb1db90d20fc674057821c11e31 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Tue, 22 Oct 2024 12:20:19 -0400 Subject: [PATCH 04/11] add back big python benchmark --- tests/python_array_optimize_big.egg | 440 ++++++++++++++++++++++++++++ 1 file changed, 440 insertions(+) create mode 100644 tests/python_array_optimize_big.egg diff --git a/tests/python_array_optimize_big.egg b/tests/python_array_optimize_big.egg new file mode 100644 index 00000000..63ddaa49 --- /dev/null +++ b/tests/python_array_optimize_big.egg @@ -0,0 +1,440 @@ +;; Old version of python array optimization that does not include CSE with let statements +;; serves as a benchmark for adding large expressions to egglog +(ruleset array_api_ruleset) +(sort Int) +(function cast_Callable__Int__Int___Int___lambda_i_____i_ (Int Int) Int) +(rewrite (cast_Callable__Int__Int___Int___lambda_i_____i_ __var__i __var___) __var__i :ruleset array_api_ruleset) +(sort TupleInt) +(function TupleInt_single (Int) TupleInt) +(sort UnstableFn_Int_Int (UnstableFn (Int) Int)) +(function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt) +(function Int___init__ (i64) Int) +(rewrite (TupleInt_single __var__i) (TupleInt___init__ (Int___init__ 1) (unstable-fn "cast_Callable__Int__Int___Int___lambda_i_____i_" __var__i)) :ruleset array_api_ruleset) +(function cast_Callable__Int___Int___lambda_i__i_ (Int) Int) +(rewrite (cast_Callable__Int___Int___lambda_i__i_ __var__i) __var__i :ruleset array_api_ruleset) +(function TupleInt_range (Int) TupleInt) +(rewrite (TupleInt_range __var__stop) (TupleInt___init__ __var__stop (unstable-fn "cast_Callable__Int___Int___lambda_i__i_")) :ruleset array_api_ruleset) +(sort Vec_Int (Vec Int)) +(function TupleInt_from_vec (Vec_Int) TupleInt) +(function index_vec_int (Vec_Int Int) Int) +(rewrite (TupleInt_from_vec __var__vec) (TupleInt___init__ (Int___init__ (vec-length __var__vec)) (unstable-fn "index_vec_int" __var__vec)) :ruleset array_api_ruleset) +(function cast_Callable__TupleInt__TupleInt__Int___Int___lambda_other__self__i__Int_if__i_<_self_length____self_i___other_i_-_self_length_____ (TupleInt TupleInt Int) Int) +(sort Boolean) +(function Int_if_ (Boolean Int Int) Int) +(function Int___lt__ (Int Int) Boolean) +(function TupleInt_length (TupleInt) Int) +(function TupleInt___getitem__ (TupleInt Int) Int) +(function Int___sub__ (Int Int) Int) +(rewrite (cast_Callable__TupleInt__TupleInt__Int___Int___lambda_other__self__i__Int_if__i_<_self_length____self_i___other_i_-_self_length_____ __var__other __var__self __var__i) (Int_if_ (Int___lt__ __var__i (TupleInt_length __var__self)) (TupleInt___getitem__ __var__self __var__i) (TupleInt___getitem__ __var__other (Int___sub__ __var__i (TupleInt_length __var__self)))) :ruleset array_api_ruleset) +(function TupleInt___add__ (TupleInt TupleInt) TupleInt) +(function Int___add__ (Int Int) Int) +(rewrite (TupleInt___add__ __var__self __var__other) (TupleInt___init__ (Int___add__ (TupleInt_length __var__self) (TupleInt_length __var__other)) (unstable-fn "cast_Callable__TupleInt__TupleInt__Int___Int___lambda_other__self__i__Int_if__i_<_self_length____self_i___other_i_-_self_length_____" __var__other __var__self)) :ruleset array_api_ruleset) +(function cast_Callable__Int__Boolean__Int___Boolean___lambda_i__acc__j__acc_|__i_==_j__ (Int Boolean Int) Boolean) +(function Boolean___or__ (Boolean Boolean) Boolean) +(function Int___eq__ (Int Int) Boolean) +(rewrite (cast_Callable__Int__Boolean__Int___Boolean___lambda_i__acc__j__acc_|__i_==_j__ __var__i __var__acc __var__j) (Boolean___or__ __var__acc (Int___eq__ __var__i __var__j)) :ruleset array_api_ruleset) +(function TupleInt_contains (TupleInt Int) Boolean) +(sort UnstableFn_Boolean_Boolean_Int (UnstableFn (Boolean Int) Boolean)) +(function TupleInt_fold_boolean (TupleInt Boolean UnstableFn_Boolean_Boolean_Int) Boolean) +(function FALSE () Boolean) +(rewrite (TupleInt_contains __var__self __var__i) (TupleInt_fold_boolean __var__self (FALSE) (unstable-fn "cast_Callable__Int__Boolean__Int___Boolean___lambda_i__acc__j__acc_|__i_==_j__" __var__i)) :ruleset array_api_ruleset) +(sort NDArray) +(function NDArray_size (NDArray) Int) +(sort UnstableFn_Int_Int_Int (UnstableFn (Int Int) Int)) +(function TupleInt_fold (TupleInt Int UnstableFn_Int_Int_Int) Int) +(function NDArray_shape (NDArray) TupleInt) +(function Int___mul__ (Int Int) Int) +(rewrite (NDArray_size __var__x) (TupleInt_fold (NDArray_shape __var__x) (Int___init__ 1) (unstable-fn "Int___mul__")) :ruleset array_api_ruleset) +(function unique_values (NDArray) NDArray) +(sort TupleValue) +(function NDArray_vector (TupleValue) NDArray) +(sort Value) +(function possible_values (Value) TupleValue) +(function NDArray_index (NDArray TupleInt) Value) +(function ALL_INDICES () TupleInt) +(rewrite (unique_values __var__a) (NDArray_vector (possible_values (NDArray_index __var__a (ALL_INDICES)))) :ruleset array_api_ruleset) +(function Value_isfinite (Value) Boolean) +(function Value_int (Int) Value) +(function TRUE () Boolean) +(rewrite (Value_isfinite (Value_int __var__i)) (TRUE) :ruleset array_api_ruleset) +(function Value_bool (Boolean) Value) +(rewrite (Value_isfinite (Value_bool __var__b)) (TRUE) :ruleset array_api_ruleset) +(sort Float) +(function Value_float (Float) Value) +(function Float___init__ (f64) Float :cost 3) +(rewrite (Value_isfinite (Value_float (Float___init__ __var__f))) (TRUE) :when ((!= __var__f NaN)) :ruleset array_api_ruleset) +(function isfinite (NDArray) NDArray) +(sort OptionalIntOrTuple) +(function sum (NDArray OptionalIntOrTuple) NDArray) +(function OptionalIntOrTuple_none () OptionalIntOrTuple) +(function NDArray_scalar (Value) NDArray) +(rewrite (isfinite (sum __var__arr (OptionalIntOrTuple_none))) (NDArray_scalar (Value_bool (Value_isfinite (NDArray_index __var__arr (ALL_INDICES))))) :ruleset array_api_ruleset) +(function assume_value_one_of (NDArray TupleValue) NDArray) +(rewrite (NDArray_shape (assume_value_one_of __var__x __var__vs)) (NDArray_shape __var__x) :ruleset array_api_ruleset) +(sort DType) +(function NDArray_dtype (NDArray) DType) +(rewrite (NDArray_dtype (assume_value_one_of __var__x __var__vs)) (NDArray_dtype __var__x) :ruleset array_api_ruleset) +(rule ((= __var__v (NDArray_index (assume_value_one_of __var__x __var__vs) __var__idx))) + ((union __var__v (NDArray_index __var__x __var__idx)) + (union (possible_values __var__v) __var__vs)) + :ruleset array_api_ruleset ) +(function assume_isfinite (NDArray) NDArray) +(rewrite (NDArray_shape (assume_isfinite __var__x)) (NDArray_shape __var__x) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_isfinite __var__x)) (NDArray_dtype __var__x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_isfinite __var__x) __var__ti) (NDArray_index __var__x __var__ti) :ruleset array_api_ruleset) +(rewrite (Value_isfinite (NDArray_index (assume_isfinite __var__x) __var__ti)) (TRUE) :ruleset array_api_ruleset) +(function assume_shape (NDArray TupleInt) NDArray) +(rewrite (NDArray_shape (assume_shape __var__x __var__shape)) __var__shape :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (assume_shape __var__x __var__shape)) (NDArray_dtype __var__x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_shape __var__x __var__shape) __var__idx) (NDArray_index __var__x __var__idx) :ruleset array_api_ruleset) +(function assume_dtype (NDArray DType) NDArray) +(rewrite (NDArray_dtype (assume_dtype __var__x __var__dtype)) __var__dtype :ruleset array_api_ruleset) +(rewrite (NDArray_shape (assume_dtype __var__x __var__dtype)) (NDArray_shape __var__x) :ruleset array_api_ruleset) +(rewrite (NDArray_index (assume_dtype __var__x __var__dtype) __var__idx) (NDArray_index __var__x __var__idx) :ruleset array_api_ruleset) +(sort IndexKey) +(function NDArray___getitem__ (NDArray IndexKey) NDArray) +(function IndexKey_int (Int) IndexKey) +(rewrite (NDArray___getitem__ __var__x (IndexKey_int __var__i)) (NDArray_scalar (NDArray_index __var__x (TupleInt_single __var__i))) :ruleset array_api_ruleset) +(sort OptionalBool) +(function reshape (NDArray TupleInt OptionalBool) NDArray) +(rule ((= __a (reshape __var__x __var__shape __var__copy))) + ((NDArray_shape __var__x) + (TupleInt_length (NDArray_shape __var__x))) + :ruleset array_api_ruleset ) +(rule ((reshape __var__x __var__shape __var__copy)) + ((TupleInt_length __var__shape) + (TupleInt___getitem__ __var__shape (Int___init__ 0))) + :ruleset array_api_ruleset ) +(rewrite (reshape __var__x __var__shape __var__copy) __var__x :when ((= (TupleInt_length (NDArray_shape __var__x)) (Int___init__ 1)) (= (TupleInt_length __var__shape) (Int___init__ 1)) (= (TupleInt___getitem__ __var__shape (Int___init__ 0)) (Int___init__ -1))) :ruleset array_api_ruleset) +(function TupleValue_length (TupleValue) Int) +(rewrite (NDArray_shape (NDArray_vector __var__vs)) (TupleInt_single (TupleValue_length __var__vs)) :ruleset array_api_ruleset) +(function Value_dtype (Value) DType) +(function TupleValue___getitem__ (TupleValue Int) Value) +(rewrite (NDArray_dtype (NDArray_vector __var__vs)) (Value_dtype (TupleValue___getitem__ __var__vs (Int___init__ 0))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_vector __var__vs) __var__ti) (TupleValue___getitem__ __var__vs (TupleInt___getitem__ __var__ti (Int___init__ 0))) :ruleset array_api_ruleset) +(function TupleInt_EMPTY () TupleInt) +(rewrite (NDArray_shape (NDArray_scalar __var__v)) (TupleInt_EMPTY) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (NDArray_scalar __var__v)) (Value_dtype __var__v) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_scalar __var__v) (TupleInt_EMPTY)) __var__v :ruleset array_api_ruleset) +(function any (NDArray) NDArray) +(function TupleValue_includes (TupleValue Value) Boolean) +(function Value_to_truthy_value (Value) Value) +(rewrite (any __var__x) (NDArray_scalar (Value_bool (TupleValue_includes (possible_values (Value_to_truthy_value (NDArray_index __var__x (ALL_INDICES)))) (Value_bool (TRUE))))) :ruleset array_api_ruleset) +(function NDArray___lt__ (NDArray NDArray) NDArray) +(function Value___lt__ (Value Value) Value) +(function broadcast_index (TupleInt TupleInt TupleInt) TupleInt) +(function broadcast_shapes (TupleInt TupleInt) TupleInt) +(rewrite (NDArray_index (NDArray___lt__ __var__x __var__y) __var__idx) (Value___lt__ (NDArray_index __var__x (broadcast_index (NDArray_shape __var__x) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx)) (NDArray_index __var__y (broadcast_index (NDArray_shape __var__y) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx))) :ruleset array_api_ruleset) +(function NDArray___truediv__ (NDArray NDArray) NDArray) +(function Value___truediv__ (Value Value) Value) +(rewrite (NDArray_index (NDArray___truediv__ __var__x __var__y) __var__idx) (Value___truediv__ (NDArray_index __var__x (broadcast_index (NDArray_shape __var__x) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx)) (NDArray_index __var__y (broadcast_index (NDArray_shape __var__y) (broadcast_shapes (NDArray_shape __var__x) (NDArray_shape __var__y)) __var__idx))) :ruleset array_api_ruleset) +(rewrite (NDArray_index (NDArray_scalar __var__v) __var__idx) __var__v :ruleset array_api_ruleset) +(function astype (NDArray DType) NDArray) +(function Value_astype (Value DType) Value) +(rewrite (NDArray_index (astype __var__x __var__dtype) __var__idx) (Value_astype (NDArray_index __var__x __var__idx) __var__dtype) :ruleset array_api_ruleset) +(relation greater_zero (Value)) +(sort TupleNDArray) +(function TupleNDArray___getitem__ (TupleNDArray Int) NDArray) +(function unique_counts (NDArray) TupleNDArray) +(rule ((= __var__v (NDArray_index (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) __var__idx))) + ((greater_zero __var__v)) + :ruleset array_api_ruleset ) +(rule ((greater_zero __var__v) + (= __var__v1 (Value_astype __var__v __var__dtype))) + ((greater_zero __var__v1)) + :ruleset array_api_ruleset ) +(rule ((= __var__v (Value_float (Float___init__ __var__f))) + (> __var__f 0.0)) + ((greater_zero __var__v)) + :ruleset array_api_ruleset ) +(rule ((= __var__v (Value_int (Int___init__ __var__i))) + (> __var__i 0)) + ((greater_zero __var__v)) + :ruleset array_api_ruleset ) +(rule ((greater_zero __var__v) + (greater_zero __var__v1) + (= __var__v2 (Value___truediv__ __var__v __var__v1))) + ((greater_zero __var__v2)) + :ruleset array_api_ruleset ) +(rule ((greater_zero __var__v) + (= __var__v1 (Value___lt__ __var__v (Value_int (Int___init__ 0))))) + ((union __var__v1 (Value_bool (FALSE)))) + :ruleset array_api_ruleset ) +(function TupleValue___init__ (Value) TupleValue) +(rewrite (possible_values (Value_bool __var__b)) (TupleValue___init__ (Value_bool __var__b)) :ruleset array_api_ruleset) +(rule ((= __var__v1 (Value_astype __var__v __var__dtype)) + (greater_zero __var__v)) + ((greater_zero __var__v1)) + :ruleset array_api_ruleset ) +(function TupleNDArray_length (TupleNDArray) Int) +(function svd (NDArray Boolean) TupleNDArray) +(rewrite (TupleNDArray_length (svd __var__x __var__full_matrices)) (Int___init__ 3) :ruleset array_api_ruleset) +(function unique_inverse (NDArray) TupleNDArray) +(rewrite (TupleNDArray_length (unique_inverse __var__x)) (Int___init__ 2) :ruleset array_api_ruleset) +(rewrite (TupleNDArray___getitem__ (unique_inverse __var__x) (Int___init__ 0)) (unique_values __var__x) :ruleset array_api_ruleset) +(function ndarray-abs (NDArray) NDArray) +(function Float_abs (Float) Float) +(rewrite (ndarray-abs (NDArray_scalar (Value_float __var__f))) (NDArray_scalar (Value_float (Float_abs __var__f))) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (unique_counts __var__x)) (Int___init__ 2) :ruleset array_api_ruleset) +(rewrite (sum (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_int (NDArray_size __var__x))) :ruleset array_api_ruleset) +(rewrite (sum (astype (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) __var__dtype) (OptionalIntOrTuple_none)) (astype (NDArray_scalar (Value_int (NDArray_size __var__x))) __var__dtype) :ruleset array_api_ruleset) +(rewrite (NDArray_dtype (astype __var__x __var__dtype)) __var__dtype :ruleset array_api_ruleset) +(function DType_float64 () DType) +(rewrite (astype (NDArray_scalar (Value_int (Int___init__ __var__i))) (DType_float64)) (NDArray_scalar (Value_float (Float___init__ (to-f64 __var__i)))) :ruleset array_api_ruleset) +(sort OptionalInt) +(function concat (TupleNDArray OptionalInt) NDArray) +(function TupleNDArray___init__ (NDArray) TupleNDArray) +(function OptionalInt_none () OptionalInt) +(rewrite (concat (TupleNDArray___init__ __var__x) (OptionalInt_none)) __var__x :ruleset array_api_ruleset) +(rewrite (unique_values (unique_values __var__x)) (unique_values __var__x) :ruleset array_api_ruleset) +(rewrite (sum (NDArray___truediv__ __var__x (NDArray_scalar __var__v)) (OptionalIntOrTuple_none)) (NDArray___truediv__ (sum __var__x (OptionalIntOrTuple_none)) (NDArray_scalar __var__v)) :ruleset array_api_ruleset) +(function NDArray_ndim (NDArray) Int) +(sort OptionalDType) +(sort OptionalDevice) +(function asarray (NDArray OptionalDType OptionalBool OptionalDevice) NDArray) +(function OptionalDevice_none () OptionalDevice) +(rewrite (NDArray_ndim (asarray __var__a __var__d __var__ob (OptionalDevice_none))) (NDArray_ndim __var__a) :ruleset array_api_ruleset) +(function OptionalDType_none () OptionalDType) +(function OptionalBool_none () OptionalBool) +(rewrite (asarray __var__a (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) __var__a :ruleset array_api_ruleset) +(function TupleNDArray___add__ (TupleNDArray TupleNDArray) TupleNDArray) +(function TupleNDArray_EMPTY () TupleNDArray) +(rewrite (TupleNDArray___add__ __var__ti (TupleNDArray_EMPTY)) __var__ti :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (TupleNDArray___init__ __var__n)) (Int___init__ 1) :ruleset array_api_ruleset) +(rewrite (TupleNDArray_length (TupleNDArray___add__ __var__ti __var__ti2)) (Int___add__ (TupleNDArray_length __var__ti) (TupleNDArray_length __var__ti2)) :ruleset array_api_ruleset) +(rewrite (NDArray_ndim __var__x) (TupleInt_length (NDArray_shape __var__x)) :ruleset array_api_ruleset) +(function NDArray_to_value (NDArray) Value) +(rewrite (NDArray_to_value __var__x) (NDArray_index __var__x (TupleInt_EMPTY)) :ruleset array_api_ruleset) +(rewrite (NDArray___truediv__ (NDArray_scalar (Value_float __var__f)) (NDArray_scalar (Value_float __var__f))) (NDArray_scalar (Value_float (Float___init__ 1.0))) :ruleset array_api_ruleset) +(function NDArray___sub__ (NDArray NDArray) NDArray) +(rewrite (NDArray___sub__ (NDArray_scalar (Value_float __var__f)) (NDArray_scalar (Value_float __var__f))) (NDArray_scalar (Value_float (Float___init__ 0.0))) :ruleset array_api_ruleset) +(function NDArray___gt__ (NDArray NDArray) NDArray) +(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ __var__fi1))) (NDArray_scalar (Value_float (Float___init__ __var__fi2)))) (NDArray_scalar (Value_bool (TRUE))) :when ((> __var__fi1 __var__fi2)) :ruleset array_api_ruleset) +(rewrite (NDArray___gt__ (NDArray_scalar (Value_float (Float___init__ __var__fi1))) (NDArray_scalar (Value_float (Float___init__ __var__fi2)))) (NDArray_scalar (Value_bool (FALSE))) :when ((<= __var__fi1 __var__fi2)) :ruleset array_api_ruleset) +(function NDArray_T (NDArray) NDArray) +(rewrite (NDArray_T (NDArray_T __var__x)) __var__x :ruleset array_api_ruleset) +(function TupleValue___add__ (TupleValue TupleValue) TupleValue) +(function TupleValue_EMPTY () TupleValue) +(rewrite (TupleValue___add__ __var__ti (TupleValue_EMPTY)) __var__ti :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue_EMPTY)) (Int___init__ 0) :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue___init__ __var__v)) (Int___init__ 1) :ruleset array_api_ruleset) +(rewrite (TupleValue_length (TupleValue___add__ __var__ti __var__ti2)) (Int___add__ (TupleValue_length __var__ti) (TupleValue_length __var__ti2)) :ruleset array_api_ruleset) +(rewrite (TupleValue___getitem__ (TupleValue___init__ __var__v) (Int___init__ 0)) __var__v :ruleset array_api_ruleset) +(rewrite (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ __var__v) __var__ti) (Int___init__ 0)) __var__v :ruleset array_api_ruleset) +(rule ((= __var__v (TupleValue___getitem__ (TupleValue___add__ (TupleValue___init__ __var__v2) __var__ti) (Int___init__ __var__k))) + (> __var__k 0)) + ((union __var__v (TupleValue___getitem__ __var__ti (Int___init__ (- __var__k 1))))) + :ruleset array_api_ruleset ) +(rewrite (TupleValue_includes (TupleValue_EMPTY) __var__v) (FALSE) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___init__ __var__v) __var__v) (TRUE) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___init__ __var__v) __var__v2) (FALSE) :when ((!= __var__v __var__v2)) :ruleset array_api_ruleset) +(rewrite (TupleValue_includes (TupleValue___add__ __var__ti __var__ti2) __var__v) (Boolean___or__ (TupleValue_includes __var__ti __var__v) (TupleValue_includes __var__ti2 __var__v)) :ruleset array_api_ruleset) +(function DType_int64 () DType) +(rewrite (Value_dtype (Value_int __var__i)) (DType_int64) :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_float __var__f)) (DType_float64) :ruleset array_api_ruleset) +(function DType_bool () DType) +(rewrite (Value_dtype (Value_bool __var__b)) (DType_bool) :ruleset array_api_ruleset) +(function Value_to_bool (Value) Boolean) +(rewrite (Value_to_bool (Value_bool __var__b)) __var__b :ruleset array_api_ruleset) +(function Value_to_int (Value) Int) +(rewrite (Value_to_int (Value_int __var__i)) __var__i :ruleset array_api_ruleset) +(rewrite (Value_to_truthy_value (Value_bool __var__b)) (Value_bool __var__b) :ruleset array_api_ruleset) +(sort IsDtypeKind) +(function isdtype (DType IsDtypeKind) Boolean) +(function DType_float32 () DType) +(function IsDtypeKind_string (String) IsDtypeKind) +(rewrite (isdtype (DType_float32) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float64) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(function DType_object () DType) +(rewrite (isdtype (DType_object) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int64) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) +(function DType_int32 () DType) +(rewrite (isdtype (DType_int32) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float32) (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float64) (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_object) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int64) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int32) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_float64) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_object) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int64) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype (DType_int32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(function IsDtypeKind_NULL () IsDtypeKind) +(rewrite (isdtype __var__d (IsDtypeKind_NULL)) (FALSE) :ruleset array_api_ruleset) +(function IsDtypeKind_dtype (DType) IsDtypeKind) +(rewrite (isdtype __var__d (IsDtypeKind_dtype __var__d)) (TRUE) :ruleset array_api_ruleset) +(function IsDtypeKind___or__ (IsDtypeKind IsDtypeKind) IsDtypeKind :cost 10) +(rewrite (isdtype __var__d (IsDtypeKind___or__ __var__k1 __var__k2)) (Boolean___or__ (isdtype __var__d __var__k1) (isdtype __var__d __var__k2)) :ruleset array_api_ruleset) +(rewrite (IsDtypeKind___or__ __var__k1 (IsDtypeKind_NULL)) __var__k1 :ruleset array_api_ruleset) +(function DType___eq__ (DType DType) Boolean) +(rewrite (DType___eq__ (DType_float64) (DType_float64)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float64) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float64) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float64) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float64) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_float64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_float32)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_float64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_int32)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_float64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_int64)) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_float64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) (DType_object)) (TRUE) :ruleset array_api_ruleset) +(function cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___ (UnstableFn_Int_Int Int) Int) +(rewrite (cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___ __var__idx_fn __var__i) (unstable-app __var__idx_fn (Int___add__ __var__i (Int___init__ 1))) :ruleset array_api_ruleset) +(rewrite (index_vec_int __var__vs (Int___init__ __var__k)) (vec-get __var__vs __var__k) :ruleset array_api_ruleset) +(rewrite (TupleInt_length (TupleInt___init__ __var__i __var__idx_fn)) __var__i :ruleset array_api_ruleset) +(rewrite (TupleInt___getitem__ (TupleInt___init__ __var__i __var__idx_fn) __var__i2) (unstable-app __var__idx_fn __var__i2) :ruleset array_api_ruleset) +(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ 0) __var__idx_fn) __var__i __var__f) __var__i :ruleset array_api_ruleset) +(rewrite (TupleInt_fold (TupleInt___init__ (Int___init__ __var__k) __var__idx_fn) __var__i __var__f) (unstable-app __var__f (TupleInt_fold (TupleInt___init__ (Int___init__ (- __var__k 1)) (unstable-fn "cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___" __var__idx_fn)) __var__i __var__f) (unstable-app __var__idx_fn (Int___init__ 0))) :when ((!= __var__k 0)) :ruleset array_api_ruleset) +(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ 0) __var__idx_fn) __var__b __var__bool_f) __var__b :ruleset array_api_ruleset) +(rewrite (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ __var__k) __var__idx_fn) __var__b __var__bool_f) (unstable-app __var__bool_f (TupleInt_fold_boolean (TupleInt___init__ (Int___init__ (- __var__k 1)) (unstable-fn "cast_Callable__UnstableFn_Int__Int___Int___Int___lambda_idx_fn__i__idx_fn_i_+_Int_1___" __var__idx_fn)) __var__b __var__bool_f) (unstable-app __var__idx_fn (Int___init__ 0))) :when ((!= __var__k 0)) :ruleset array_api_ruleset) +(function bottom_indexing (Int) Int) +(rewrite (TupleInt_EMPTY) (TupleInt___init__ (Int___init__ 0) (unstable-fn "bottom_indexing")) :ruleset array_api_ruleset) +(rewrite (Float_abs (Float___init__ __var__f)) (Float___init__ __var__f) :when ((>= __var__f 0.0)) :ruleset array_api_ruleset) +(rewrite (Float_abs (Float___init__ __var__f)) (Float___init__ (neg __var__f)) :when ((< __var__f 0.0)) :ruleset array_api_ruleset) +(function Float_rational (Rational) Float :cost 2) +(rewrite (Float___init__ __var__f) (Float_rational (rational (to-i64 __var__f) 1)) :when ((= (to-f64 (to-i64 __var__f)) __var__f)) :ruleset array_api_ruleset) +(function Float_from_int (Int) Float) +(rewrite (Float_from_int (Int___init__ __var__i)) (Float_rational (rational __var__i 1)) :ruleset array_api_ruleset) +(function Float___add__ (Float Float) Float) +(rewrite (Float___add__ (Float___init__ __var__f) (Float___init__ __var__f2)) (Float___init__ (+ __var__f __var__f2)) :ruleset array_api_ruleset) +(function Float___sub__ (Float Float) Float) +(rewrite (Float___sub__ (Float___init__ __var__f) (Float___init__ __var__f2)) (Float___init__ (- __var__f __var__f2)) :ruleset array_api_ruleset) +(function Float___mul__ (Float Float) Float) +(rewrite (Float___mul__ (Float___init__ __var__f) (Float___init__ __var__f2)) (Float___init__ (* __var__f __var__f2)) :ruleset array_api_ruleset) +(function Float___truediv__ (Float Float) Float) +(rewrite (Float___truediv__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (/ __var__r __var__r1)) :ruleset array_api_ruleset) +(rewrite (Float___add__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (+ __var__r __var__r1)) :ruleset array_api_ruleset) +(rewrite (Float___sub__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (- __var__r __var__r1)) :ruleset array_api_ruleset) +(rewrite (Float___mul__ (Float_rational __var__r) (Float_rational __var__r1)) (Float_rational (* __var__r __var__r1)) :ruleset array_api_ruleset) +(rewrite (Int___eq__ (Int___init__ __var__i) (Int___init__ __var__i)) (TRUE) :ruleset array_api_ruleset) +(rule ((= __var__r (Int___eq__ (Int___init__ __var__i) (Int___init__ __var__j))) + (!= __var__i __var__j)) + ((union __var__r (FALSE))) + :ruleset array_api_ruleset ) +(function Int___ge__ (Int Int) Boolean) +(rewrite (Int___ge__ (Int___init__ __var__i) (Int___init__ __var__i)) (TRUE) :ruleset array_api_ruleset) +(rule ((= __var__r (Int___ge__ (Int___init__ __var__i) (Int___init__ __var__j))) + (> __var__i __var__j)) + ((union __var__r (TRUE))) + :ruleset array_api_ruleset ) +(rule ((= __var__r (Int___ge__ (Int___init__ __var__i) (Int___init__ __var__j))) + (< __var__i __var__j)) + ((union __var__r (FALSE))) + :ruleset array_api_ruleset ) +(rewrite (Int___lt__ (Int___init__ __var__i) (Int___init__ __var__i)) (FALSE) :ruleset array_api_ruleset) +(rule ((= __var__r (Int___lt__ (Int___init__ __var__i) (Int___init__ __var__j))) + (< __var__i __var__j)) + ((union __var__r (TRUE))) + :ruleset array_api_ruleset ) +(rule ((= __var__r (Int___lt__ (Int___init__ __var__i) (Int___init__ __var__j))) + (> __var__i __var__j)) + ((union __var__r (FALSE))) + :ruleset array_api_ruleset ) +(function Int___gt__ (Int Int) Boolean) +(rewrite (Int___gt__ (Int___init__ __var__i) (Int___init__ __var__i)) (FALSE) :ruleset array_api_ruleset) +(rule ((= __var__r (Int___gt__ (Int___init__ __var__i) (Int___init__ __var__j))) + (> __var__i __var__j)) + ((union __var__r (TRUE))) + :ruleset array_api_ruleset ) +(rule ((= __var__r (Int___gt__ (Int___init__ __var__i) (Int___init__ __var__j))) + (< __var__i __var__j)) + ((union __var__r (FALSE))) + :ruleset array_api_ruleset ) +(function Int_i64 (Int) i64) +(rule ((= __var__o (Int___init__ __var__j))) + ((set (Int_i64 __var__o) __var__j)) + :ruleset array_api_ruleset ) +(rewrite (Int___add__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (+ __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___sub__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (- __var__i __var__j)) :ruleset array_api_ruleset) +(rewrite (Int___mul__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (* __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___truediv__ (Int Int) Int) +(rewrite (Int___truediv__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (/ __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___mod__ (Int Int) Int) +(rewrite (Int___mod__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (% __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___and__ (Int Int) Int) +(rewrite (Int___and__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (& __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___or__ (Int Int) Int) +(rewrite (Int___or__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (| __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___xor__ (Int Int) Int) +(rewrite (Int___xor__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (^ __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___lshift__ (Int Int) Int) +(rewrite (Int___lshift__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (<< __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___rshift__ (Int Int) Int) +(rewrite (Int___rshift__ (Int___init__ __var__i) (Int___init__ __var__j)) (Int___init__ (>> __var__i __var__j)) :ruleset array_api_ruleset) +(function Int___invert__ (Int) Int) +(rewrite (Int___invert__ (Int___init__ __var__i)) (Int___init__ (not-i64 __var__i)) :ruleset array_api_ruleset) +(rewrite (Int_if_ (TRUE) __var__o __var__b) __var__o :ruleset array_api_ruleset) +(rewrite (Int_if_ (FALSE) __var__o __var__b) __var__b :ruleset array_api_ruleset) +(function Boolean_bool (Boolean) bool) +(rule ((= __var__x (TRUE))) + ((set (Boolean_bool __var__x) true)) + :ruleset array_api_ruleset ) +(rule ((= __var__x (FALSE))) + ((set (Boolean_bool __var__x) false)) + :ruleset array_api_ruleset ) +(rewrite (Boolean___or__ (TRUE) __var__x) (TRUE) :ruleset array_api_ruleset) +(rewrite (Boolean___or__ (FALSE) __var__x) __var__x :ruleset array_api_ruleset) +(function Boolean___and__ (Boolean Boolean) Boolean) +(rewrite (Boolean___and__ (TRUE) __var__x) __var__x :ruleset array_api_ruleset) +(rewrite (Boolean___and__ (FALSE) __var__x) (FALSE) :ruleset array_api_ruleset) +(function Boolean_if_int (Boolean Int Int) Int) +(rewrite (Boolean_if_int (TRUE) __var__i __var__j) __var__i :ruleset array_api_ruleset) +(rewrite (Boolean_if_int (FALSE) __var__i __var__j) __var__j :ruleset array_api_ruleset) +(function Boolean___invert__ (Boolean) Boolean) +(rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) +(rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) +(ruleset ruleset_5218390416) +(function NDArray___eq__ (NDArray NDArray) NDArray) +(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse __var__x) (Int___init__ 1)) (NDArray_scalar (Value_int __var__i))) (NDArray___eq__ __var__x (NDArray_scalar (NDArray_index (unique_values __var__x) (TupleInt_from_vec (vec-of __var__i))))) :subsume :ruleset ruleset_5218390416) +(function count_values (NDArray NDArray) TupleValue :unextractable) +(rewrite (TupleNDArray___getitem__ (unique_counts __var__x) (Int___init__ 1)) (NDArray_vector (count_values __var__x (unique_values __var__x))) :subsume :ruleset ruleset_5218390416) +(rewrite (count_values __var__x (NDArray_vector (TupleValue___add__ (TupleValue___init__ __var__v) __var__tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ __var__x (NDArray_scalar __var__v)) (OptionalIntOrTuple_none)))) (count_values __var__x (NDArray_vector __var__tv))) :subsume :ruleset ruleset_5218390416) +(rewrite (count_values __var__x (NDArray_vector (TupleValue___init__ __var__v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ __var__x (NDArray_scalar __var__v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_5218390416) +(function std (NDArray OptionalIntOrTuple) NDArray) +(sort IntOrTuple) +(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) +(function IntOrTuple_int (Int) IntOrTuple) +(function ndarray-sqrt (NDArray) NDArray) +(function mean (NDArray OptionalIntOrTuple Boolean) NDArray) +(function square (NDArray) NDArray) +(rewrite (std __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i))) (ndarray-sqrt (mean (square (NDArray___sub__ __var__x (mean __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (FALSE))) :subsume :ruleset ruleset_5218390416) +(rewrite (mean __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (FALSE)) (NDArray___truediv__ (sum __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape __var__x) __var__i)))) :subsume :ruleset ruleset_5218390416) +(function expand_dims (NDArray Int) NDArray) +(rewrite (mean __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum __var__x (OptionalIntOrTuple_some (IntOrTuple_int __var__i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape __var__x) __var__i)))) __var__i) :subsume :ruleset ruleset_5218390416) +(function NDArray___matmul__ (NDArray NDArray) NDArray) +(function NDArray_var (String) NDArray :cost 200) +(function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) +(function zeros (TupleInt OptionalDType OptionalDevice) NDArray) +(function OptionalDType_some (DType) OptionalDType) +(sort Device) +(function OptionalDevice_some (Device) OptionalDevice) +(function NDArray_device (NDArray) Device) +(sort MultiAxisIndexKey) +(function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) +(sort MultiAxisIndexKeyItem) +(sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) +(function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) +(function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) +(sort Slice) +(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) +(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) +(function IndexKey_ndarray (NDArray) IndexKey) +(function NDArray___mul__ (NDArray NDArray) NDArray) +(function OptionalInt_some (Int) OptionalInt) +(function IndexKey_slice (Slice) IndexKey) +(simplify (saturate (seq (run array_api_ruleset) (run ruleset_5218390416))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))))) (NDArray___matmul__ (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))))) (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0)) (Int___init__ 1)))))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE)))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none)))))) (FALSE)) (Int___init__ 2))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0)) (Int___init__ 1)))))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE)))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none)))))) (FALSE)) (Int___init__ 1)) (NDArray___mul__ (NDArray_scalar (Value_float (Float___init__ 0.0001))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0)) (Int___init__ 1)))))))) (NDArray_T (NDArray___sub__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE)))))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 2)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (IndexKey_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ (Float___init__ 1.0) (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none)))) (Int___init__ 0))))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (NDArray___truediv__ (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (NDArray___setitem__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (IndexKey_ndarray (NDArray___eq__ (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 0)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 1)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (NDArray___getitem__ (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalInt_none))) (IndexKey_int (Int___init__ 2)))))) (NDArray___getitem__ (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 0)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 0)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 1)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 1)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))))) (mean (NDArray___getitem__ (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (IndexKey_ndarray (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (TupleInt_from_vec (vec-of (Int___init__ -1))) (OptionalBool_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (NDArray_scalar (Value_int (Int___init__ 2)))))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0))) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int (Int___init__ 2)) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))))))))))) (OptionalInt_some (Int___init__ 0))) (OptionalIntOrTuple_some (IntOrTuple_int (Int___init__ 0)))) (NDArray_scalar (Value_int (Int___init__ 0))))) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (FALSE)) (Int___init__ 1)) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none)))))) (FALSE)) (Int___init__ 1)) (IndexKey_int (Int___init__ 0))))) (DType_int32)) (OptionalIntOrTuple_none))))) (OptionalInt_none))))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none))) (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_some (Int___init__ 2)) (OptionalInt_none)))))))) From bc583690df9f7ccd4945fb05b6fc08605c8d69f8 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Tue, 22 Oct 2024 12:20:55 -0400 Subject: [PATCH 05/11] switch back to running all benchmarks --- benches/example_benchmarks.rs | 25 +++++++++---------------- 1 file changed, 9 insertions(+), 16 deletions(-) diff --git a/benches/example_benchmarks.rs b/benches/example_benchmarks.rs index 548942c7..e733986a 100644 --- a/benches/example_benchmarks.rs +++ b/benches/example_benchmarks.rs @@ -1,19 +1,6 @@ use codspeed_criterion_compat::{criterion_group, criterion_main, Criterion}; use egglog::EGraph; -// Only benchmark longer running examples, -// because many of the short ones have too much variance due to the allocator being non deterministic. -// https://github.com/oxc-project/backlog/issues/89 - -const BENCHMARKS: &[&str] = &[ - "eggcc-extraction", - "math-microbenchmark", - "herbie", - "typeinfer", - "lambda", - "python_array_optimize", -]; - fn run_example(filename: &str, program: &str) { EGraph::default() .parse_and_run_program(Some(filename.to_owned()), program) @@ -21,10 +8,16 @@ fn run_example(filename: &str, program: &str) { } pub fn criterion_benchmark(c: &mut Criterion) { - for name in BENCHMARKS { - let filename = format!("tests/{}.egg", name); + for entry in glob::glob("tests/**/*.egg").unwrap() { + let path = entry.unwrap().clone(); + let path_string = path.to_string_lossy().to_string(); + if path_string.contains("fail-typecheck") { + continue; + } + let name = path.file_stem().unwrap().to_string_lossy().to_string(); + let filename = path.to_string_lossy().to_string(); let program = std::fs::read_to_string(&filename).unwrap(); - c.bench_function(name, |b| b.iter(|| run_example(&filename, &program))); + c.bench_function(&name, |b| b.iter(|| run_example(&filename, &program))); } } From 9ab7c13aad4e40132c45bc355c465c13178c1773 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Tue, 22 Oct 2024 12:29:05 -0400 Subject: [PATCH 06/11] Add documentation about ignoring benchmarks --- README.md | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/README.md b/README.md index f0991b8e..d2eb3a84 100644 --- a/README.md +++ b/README.md @@ -124,3 +124,17 @@ These primitives are only defined when the result itself is a pure rational. Use double quotes to get a quote: `"Foo "" Bar"` is `Foo " Bar`. No primitives defined. + +## Benchmarks + +We run all of our "examples" [as benchmarks in codspeed](https://codspeed.io/egraphs-good/egglog). These are in CI +for every commit in main and for all PRs. It will run the examples with extra instrumentation added so that it can +capture a single trace of the CPU interactions ([src](https://docs.codspeed.io/features/understanding-the-metrics/)): + +> CodSpeed instruments your benchmarks to measure the performance of your code. A benchmark will be run only once and the CPU behavior will be simulated. This ensures that the measurement is as accurate as possible, taking into account not only the instructions executed but also the cache and memory access patterns. The simulation gives us an equivalent of the CPU cycles that includes cache and memory access. + +Since many of the shorter running benchmarks have unstable timings due to non deterministic behavior, we ["ignore"](https://docs.codspeed.io/features/ignoring-benchmarks/) +them in codspeed. That way, we still capture their performance, but their timings don't show up in our reports by default. + +We use 50ms as our cutoff currently, any benchmarks shorter than that are ignored. This number was selected to try to ignore +any benchmarks with have changes > 1% when they haven't been modified. From 43c55d071f8e2275debdb00281294c4e46d9a3b7 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Wed, 23 Oct 2024 21:20:30 -0400 Subject: [PATCH 07/11] Updated to new python version --- tests/python_array_optimize.egg | 1031 +++++++++++++------------------ 1 file changed, 430 insertions(+), 601 deletions(-) diff --git a/tests/python_array_optimize.egg b/tests/python_array_optimize.egg index 52f9e1e3..0e2b1bbb 100644 --- a/tests/python_array_optimize.egg +++ b/tests/python_array_optimize.egg @@ -1,9 +1,8 @@ -;; created by running test_source_optimized and removing the PyObject functions -(push 1) +;; created by `python python/tests/test_array_api.py` (push 1) (sort DType) (function DType_float64 () DType) -(let %__expr_-458881301673629543 (DType_float64)) +(let %__expr_2565664028803046192 (DType_float64)) (sort Boolean) (function DType___eq__ (DType DType) Boolean) (sort NDArray) @@ -17,7 +16,7 @@ (sort Vec_Int (Vec Int)) (function TupleInt_from_vec (Vec_Int) TupleInt) (function Int___init__ (i64) Int) -(DType___eq__ %__expr_-458881301673629543 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) +(DType___eq__ %__expr_2565664028803046192 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) (ruleset array_api_ruleset) (function i-Int__-Int_i (Int Int) Int) (rewrite (i-Int__-Int_i i _) i :ruleset array_api_ruleset) @@ -54,22 +53,6 @@ (rewrite (f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__ f self i) (unstable-app f (TupleInt___getitem__ self i)) :ruleset array_api_ruleset) (function TupleInt_map (TupleInt UnstableFn_Int_Int) TupleInt :cost 100) (rewrite (TupleInt_map self f) (TupleInt___init__ (TupleInt_length self) (unstable-fn "f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__" f self)) :ruleset array_api_ruleset) -(function i-TupleInt__-Int_i (TupleInt Int) TupleInt) -(rewrite (i-TupleInt__-Int_i i _) i :ruleset array_api_ruleset) -(sort TupleTupleInt) -(function TupleTupleInt_single (TupleInt) TupleTupleInt) -(sort UnstableFn_TupleInt_Int (UnstableFn (Int) TupleInt)) -(function TupleTupleInt___init__ (Int UnstableFn_TupleInt_Int) TupleTupleInt) -(rewrite (TupleTupleInt_single i) (TupleTupleInt___init__ (Int___init__ 1) (unstable-fn "i-TupleInt__-Int_i" i)) :ruleset array_api_ruleset) -(function TupleTupleInt_from_vec (Vec_Int) TupleInt) -(rewrite (TupleTupleInt_from_vec vec) (TupleInt___init__ (Int___init__ (vec-length vec)) (unstable-fn "index_vec_int" vec)) :ruleset array_api_ruleset) -(function TupleInt_if_ (Boolean TupleInt TupleInt) TupleInt) -(function TupleTupleInt_length (TupleTupleInt) Int) -(function TupleTupleInt___getitem__ (TupleTupleInt Int) TupleInt) -(function other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ (TupleTupleInt TupleTupleInt Int) TupleInt) -(rewrite (other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ other self i) (TupleInt_if_ (Int___lt__ i (TupleTupleInt_length self)) (TupleTupleInt___getitem__ self i) (TupleTupleInt___getitem__ other (Int___sub__ i (TupleTupleInt_length self)))) :ruleset array_api_ruleset) -(function TupleTupleInt___add__ (TupleTupleInt TupleTupleInt) TupleTupleInt) -(rewrite (TupleTupleInt___add__ self other) (TupleTupleInt___init__ (Int___add__ (TupleTupleInt_length self) (TupleTupleInt_length other)) (unstable-fn "other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____" other self)) :ruleset array_api_ruleset) (function NDArray_size (NDArray) Int) (sort UnstableFn_Int_Int_Int (UnstableFn (Int Int) Int)) (function TupleInt_fold (TupleInt Int UnstableFn_Int_Int_Int) Int) @@ -205,7 +188,7 @@ (rewrite (sum (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_int (NDArray_size x))) :ruleset array_api_ruleset) (rewrite (sum (astype (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) dtype) (OptionalIntOrTuple_none)) (astype (NDArray_scalar (Value_int (NDArray_size x))) dtype) :ruleset array_api_ruleset) (rewrite (NDArray_dtype (astype x dtype)) dtype :ruleset array_api_ruleset) -(rewrite (astype (NDArray_scalar (Value_int (Int___init__ i))) %__expr_-458881301673629543) (NDArray_scalar (Value_float (Float___init__ (to-f64 i)))) :ruleset array_api_ruleset) +(rewrite (astype (NDArray_scalar (Value_int (Int___init__ i))) %__expr_2565664028803046192) (NDArray_scalar (Value_float (Float___init__ (to-f64 i)))) :ruleset array_api_ruleset) (sort OptionalInt) (function concat (TupleNDArray OptionalInt) NDArray) (function TupleNDArray___init__ (NDArray) TupleNDArray) @@ -260,7 +243,7 @@ (rewrite (TupleValue_includes (TupleValue___add__ ti ti2) v) (Boolean___or__ (TupleValue_includes ti v) (TupleValue_includes ti2 v)) :ruleset array_api_ruleset) (function DType_int64 () DType) (rewrite (Value_dtype (Value_int i)) (DType_int64) :ruleset array_api_ruleset) -(rewrite (Value_dtype (Value_float f)) %__expr_-458881301673629543 :ruleset array_api_ruleset) +(rewrite (Value_dtype (Value_float f)) %__expr_2565664028803046192 :ruleset array_api_ruleset) (function DType_bool () DType) (rewrite (Value_dtype (Value_bool b)) (DType_bool) :ruleset array_api_ruleset) (function Value_to_bool (Value) Boolean) @@ -273,19 +256,19 @@ (function DType_float32 () DType) (function IsDtypeKind_string (String) IsDtypeKind) (rewrite (isdtype (DType_float32) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) -(rewrite (isdtype %__expr_-458881301673629543 (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_2565664028803046192 (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) (function DType_object () DType) (rewrite (isdtype (DType_object) (IsDtypeKind_string "integral")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int64) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) (function DType_int32 () DType) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "integral")) (TRUE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_float32) (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) -(rewrite (isdtype %__expr_-458881301673629543 (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_2565664028803046192 (IsDtypeKind_string "real floating")) (TRUE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_object) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int64) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "real floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_float32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) -(rewrite (isdtype %__expr_-458881301673629543 (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) +(rewrite (isdtype %__expr_2565664028803046192 (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_object) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int64) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) (rewrite (isdtype (DType_int32) (IsDtypeKind_string "complex floating")) (FALSE) :ruleset array_api_ruleset) @@ -296,27 +279,27 @@ (function IsDtypeKind___or__ (IsDtypeKind IsDtypeKind) IsDtypeKind :cost 10) (rewrite (isdtype d (IsDtypeKind___or__ k1 k2)) (Boolean___or__ (isdtype d k1) (isdtype d k2)) :ruleset array_api_ruleset) (rewrite (IsDtypeKind___or__ k1 (IsDtypeKind_NULL)) k1 :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_-458881301673629543 %__expr_-458881301673629543) (TRUE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_float32)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_int32)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_int64)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ %__expr_-458881301673629543 (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_float32) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_2565664028803046192 %__expr_2565664028803046192) (TRUE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_2565664028803046192 (DType_float32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_2565664028803046192 (DType_int32)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_2565664028803046192 (DType_int64)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ %__expr_2565664028803046192 (DType_object)) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_float32) %__expr_2565664028803046192) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_float32)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_float32) (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_int32) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int32) %__expr_2565664028803046192) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_float32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_int32)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_int64)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int32) (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_int64) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_int64) %__expr_2565664028803046192) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_float32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_int64)) (TRUE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_int64) (DType_object)) (FALSE) :ruleset array_api_ruleset) -(rewrite (DType___eq__ (DType_object) %__expr_-458881301673629543) (FALSE) :ruleset array_api_ruleset) +(rewrite (DType___eq__ (DType_object) %__expr_2565664028803046192) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_float32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_int32)) (FALSE) :ruleset array_api_ruleset) (rewrite (DType___eq__ (DType_object) (DType_int64)) (FALSE) :ruleset array_api_ruleset) @@ -333,6 +316,7 @@ (sort UnstableFn_Boolean_Int (UnstableFn (Int) Boolean)) (function TupleInt_filter (TupleInt UnstableFn_Boolean_Int) TupleInt :cost 100) (rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ 0) idx_fn) filter_f) (TupleInt___init__ (Int___init__ 0) idx_fn) :ruleset array_api_ruleset) +(function TupleInt_if_ (Boolean TupleInt TupleInt) TupleInt) (rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ k) idx_fn) filter_f) (TupleInt_if_ (unstable-app filter_f (unstable-app idx_fn (Int___init__ k))) (TupleInt___add__ (TupleInt_single (unstable-app idx_fn (Int___init__ k))) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) :when ((!= k 0)) :ruleset array_api_ruleset) (function bottom_indexing (Int) Int) (rewrite (TupleInt_EMPTY) (TupleInt___init__ (Int___init__ 0) (unstable-fn "bottom_indexing")) :ruleset array_api_ruleset) @@ -433,156 +417,160 @@ (rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) (rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) (run-schedule (saturate (run array_api_ruleset))) -(extract (DType___eq__ %__expr_-458881301673629543 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) 0) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) +(extract (Boolean_bool (DType___eq__ %__expr_2565664028803046192 (NDArray_dtype (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) 0) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) 0) -(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1))) 0) +(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) 0) -(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(extract (Boolean_bool (Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3))) 0) +(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) 0) -(let %__expr_4540085663845350143 (OptionalBool_none)) -(let %__expr_7530421370362517486 (OptionalDType_none)) -(let %__expr_-4082425080216461620 (OptionalDevice_none)) -(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) +(extract (Boolean_bool (isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL))))) 0) +(let %__expr_-3674731718720096444 (OptionalDevice_none)) +(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) 0) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Boolean_bool (Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none)))))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2)) 0) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) +(extract (Boolean_bool (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (Int___init__ 2))) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) 0) -(let %__expr_7773480733401006441 (Int___init__ 1)) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) 0) -(Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (Int___init__ 3)) +(extract (Boolean_bool (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1))) 0) +(Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444)) (Int___init__ 3)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (Int___init__ 3)) 0) -(let %__expr_-7750395295943998052 (Int___init__ 2)) -(Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) +(extract (Boolean_bool (Int___ge__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444)) (Int___init__ 3))) 0) +(let %__expr_2305922367412063525 (Int___init__ 2)) +(Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444)) %__expr_2305922367412063525) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) 0) -(TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444)) %__expr_2305922367412063525)) 0) +(TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) -(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444)))) 0) +(let %__expr_2091604061025500396 (OptionalDType_none)) +(let %__expr_-4122594143565222517 (OptionalBool_none)) +(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) 0) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Boolean_bool (isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL))))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) -(TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) 0) +(TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) -(let %__expr_871583677900651896 (Int___init__ 0)) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)))) 0) +(let %__expr_-7942988945600443502 (Int___init__ 0)) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) -(let %__expr_3713283541138310173 (Int___init__ 150)) -(Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(extract (Boolean_bool (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) 0) +(let %__expr_-3520630818915629594 (Int___init__ 150)) +(Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) +(extract (Boolean_bool (Int___gt__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) 0) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Boolean_bool (Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) -(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) 0) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_2305922367412063525) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) 0) -(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_2305922367412063525)) 0) +(let %__expr_3300329792522185471 (Int___init__ 1)) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) 0) -(TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471)) 0) +(TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) -(TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)))) 0) +(TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) 0) -(DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (DType_object)) +(extract (Int_i64 (TupleInt___getitem__ (NDArray_shape (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502)) 0) +(DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (DType_object)) (run-schedule (saturate (run array_api_ruleset))) -(extract (DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (DType_object)) 0) -(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) +(extract (Boolean_bool (DType___eq__ (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (DType_object))) 0) +(Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_2305922367412063525) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_-7750395295943998052) 0) -(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind_string "real floating")) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_2305922367412063525)) 0) +(isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (IsDtypeKind_string "real floating")) (run-schedule (saturate (run array_api_ruleset))) -(extract (isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (IsDtypeKind_string "real floating")) 0) -(Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) %__expr_871583677900651896) %__expr_-7750395295943998052) +(extract (Boolean_bool (isdtype (NDArray_dtype (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (IsDtypeKind_string "real floating"))) 0) +(Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) %__expr_-7942988945600443502) %__expr_2305922367412063525) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) %__expr_871583677900651896) %__expr_-7750395295943998052) 0) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Boolean_bool (Int___gt__ (TupleInt___getitem__ (NDArray_shape (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) %__expr_-7942988945600443502) %__expr_2305922367412063525)) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) 0) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)) 0) -(TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(extract (Boolean_bool (Int___eq__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502))) 0) +(TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) -(TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) +(extract (Int_i64 (TupleNDArray_length (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)))) 0) +(TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896) 0) -(let %__expr_-2697243261424336135 (Value_int (Int___init__ 0))) -(Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_-2697243261424336135))))) +(extract (Int_i64 (TupleInt___getitem__ (NDArray_shape (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502)) 0) +(let %__expr_478758938070675600 (Value_int (Int___init__ 0))) +(Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_478758938070675600))))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_-2697243261424336135))))) 0) -(Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) +(extract (Boolean_bool (Value_to_bool (NDArray_to_value (any (NDArray___lt__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (NDArray_scalar %__expr_478758938070675600)))))) 0) +(Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001)))))) 0) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441)) +(extract (Boolean_bool (Value_to_bool (NDArray_to_value (NDArray___gt__ (ndarray-abs (NDArray___sub__ (sum (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) (OptionalIntOrTuple_none)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (NDArray_scalar (Value_float (Float___init__ 0.00001))))))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441)) 0) -(Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) %__expr_-7750395295943998052) +(extract (Boolean_bool (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471))) 0) +(Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471) %__expr_2305922367412063525) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) %__expr_-7750395295943998052) 0) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Boolean_bool (Int___lt__ (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471) %__expr_2305922367412063525)) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) -(TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) 0) +(TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) 0) -(TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) %__expr_871583677900651896) +(extract (Int_i64 (TupleNDArray_length (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)))) 0) +(TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502)) %__expr_-7942988945600443502) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) %__expr_871583677900651896) 0) -(NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) +(extract (Int_i64 (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502)) %__expr_-7942988945600443502)) 0) +(NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) 0) -(let %__expr_8044523218021318818 (Value_int %__expr_7773480733401006441)) -(let %__expr_-8852798553151442138 (Value_int (Int___init__ 2))) -(let %__expr_7039565710158856600 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) -(let %__expr_9189154200368037574 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) -(let %__expr_4948775059192706173 (NDArray_scalar %__expr_-2697243261424336135)) -(let %__expr_4396021293866471344 (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441)) -(let %__expr_4942768151788412671 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Int_i64 (NDArray_size (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none))))) 0) +(let %__expr_-8664633617851008504 (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) +(let %__expr_-8475266842302933511 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(let %__expr_-6765600943420531488 (Value_int (Int___init__ 2))) +(let %__expr_-5200143055942215294 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) +(let %__expr_-8613124149453883591 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) +(let %__expr_-622493510873225930 (Value_int (Int___init__ 1))) +(let %__expr_-3283806677371465170 (NDArray_scalar %__expr_478758938070675600)) +(let %__expr_-2704792733181451377 (TupleNDArray___getitem__ %__expr_-8664633617851008504 %__expr_3300329792522185471)) (sort MultiAxisIndexKeyItem) (sort Slice) (function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) (function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) -(let %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) -(let %__expr_6308258715381497904 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) +(let %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) (sort IntOrTuple) (function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) (function IntOrTuple_int (Int) IntOrTuple) -(let %__expr_-6383682287120045686 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896))) +(let %__expr_1913846200346632608 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-7942988945600443502))) +(let %__expr_6185990579165616281 (FALSE)) +(let %__expr_-1113027338580586528 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) (sort MultiAxisIndexKey) (function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) (sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) (function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) (function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) -(let %__expr_811525261394403186 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7750395295943998052) %__expr_4574319845551051569)))) -(let %__expr_-7672334842805227 (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) +(let %__expr_7180621784385959982 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_3300329792522185471) %__expr_-8637348939342624324)))) +(let %__expr_-4323225966908438009 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7942988945600443502) %__expr_-8637348939342624324)))) +(let %__expr_483064111004974365 (Float___init__ 1.0)) +(let %__expr_-143818766793636568 (OptionalInt_none)) (function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) (function zeros (TupleInt OptionalDType OptionalDevice) NDArray) (function OptionalDType_some (DType) OptionalDType) @@ -592,75 +580,70 @@ (function mean (NDArray OptionalIntOrTuple Boolean) NDArray) (function IndexKey_ndarray (NDArray) IndexKey) (function NDArray___eq__ (NDArray NDArray) NDArray) -(let %__expr_4481754591798653144 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device %__expr_7039565710158856600))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 %__expr_4948775059192706173))) %__expr_-6383682287120045686 (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_8044523218021318818)))) %__expr_-6383682287120045686 (FALSE))) %__expr_811525261394403186 (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_-8852798553151442138)))) %__expr_-6383682287120045686 (FALSE)))) -(let %__expr_-907407888652541496 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569)))) -(let %__expr_5801044028809884995 (FALSE)) -(let %__expr_-6354454737791578173 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))) -(let %__expr_6624227409259184391 (Float___init__ 1.0)) +(let %__expr_7843526591494960327 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ (unique_inverse (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) (OptionalDType_some (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (OptionalDevice_some (NDArray_device %__expr_-8613124149453883591))) %__expr_-4323225966908438009 (mean (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-2704792733181451377 %__expr_-3283806677371465170))) %__expr_1913846200346632608 %__expr_6185990579165616281)) %__expr_7180621784385959982 (mean (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-2704792733181451377 (NDArray_scalar %__expr_-622493510873225930)))) %__expr_1913846200346632608 %__expr_6185990579165616281)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_2305922367412063525) %__expr_-8637348939342624324))) (mean (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-2704792733181451377 (NDArray_scalar %__expr_-6765600943420531488)))) %__expr_1913846200346632608 %__expr_6185990579165616281))) +(let %__expr_-954298076671303615 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_2305922367412063525) %__expr_-8637348939342624324)))) +(function std (NDArray OptionalIntOrTuple) NDArray) (function OptionalInt_some (Int) OptionalInt) -(let %__expr_-7185048337851367263 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_871583677900651896))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-907407888652541496))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_7773480733401006441))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-6354454737791578173))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_-7750395295943998052))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_811525261394403186))))) (OptionalInt_some %__expr_871583677900651896))) +(let %__expr_3049932803880609425 (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_-7942988945600443502))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-4323225966908438009))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_3300329792522185471))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_7180621784385959982))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_2305922367412063525))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-954298076671303615))))) (OptionalInt_some %__expr_-7942988945600443502)) %__expr_1913846200346632608)) +(let %__expr_-671931098406187877 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_-7942988945600443502))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-4323225966908438009))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_3300329792522185471))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_7180621784385959982))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_2305922367412063525))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-954298076671303615))))) (OptionalInt_some %__expr_-7942988945600443502))) (function NDArray___mul__ (NDArray NDArray) NDArray) (function ndarray-sqrt (NDArray) NDArray) -(function std (NDArray OptionalIntOrTuple) NDArray) -(TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) +(TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281)) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) 0) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) +(extract (Int_i64 (TupleNDArray_length (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281))) 0) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441) 0) -(let %__expr_-8392276577365791252 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) -(let %__expr_-9127408350583128512 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)) -(let %__expr_4873895522838289209 (OptionalInt_none)) -(let %__expr_1166113493069915661 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995) %__expr_7773480733401006441)) -(let %__expr_-2809124111957918512 (std %__expr_-7185048337851367263 %__expr_-6383682287120045686)) -(let %__expr_-1704542949091505620 (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) -(let %__expr_-2471947197794367259 (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_1166113493069915661 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_4873895522838289209)) -(let %__expr_5778560295018625428 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) +(extract (Boolean_bool (Int___eq__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471)) 0) +(let %__expr_-1459297673498492435 (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281) %__expr_3300329792522185471) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_-143818766793636568)) +(let %__expr_-5540247018246576406 (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) +(let %__expr_-4744074158064860754 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281) %__expr_3300329792522185471)) +(let %__expr_2307122782736907488 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) +(let %__expr_4799320818084596281 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281)) (function NDArray___matmul__ (NDArray NDArray) NDArray) (function IndexKey_slice (Slice) IndexKey) -(TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995)) +(TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) %__expr_-5540247018246576406) (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471))))))) (NDArray_T (NDArray___sub__ %__expr_7843526591494960327 (NDArray___matmul__ %__expr_-5540247018246576406 %__expr_7843526591494960327))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_4799320818084596281 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-1459297673498492435) %__expr_-8637348939342624324)))) %__expr_2307122782736907488)) (NDArray___getitem__ %__expr_-4744074158064860754 (IndexKey_slice %__expr_-1459297673498492435)))) %__expr_6185990579165616281)) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none)))) %__expr_871583677900651896) %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441) (NDArray_dtype (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995)) 0) -(TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none))))) +(extract (Int_i64 (TupleNDArray_length (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)))) %__expr_-5540247018246576406) (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502) %__expr_3300329792522185471))))))) (NDArray_T (NDArray___sub__ %__expr_7843526591494960327 (NDArray___matmul__ %__expr_-5540247018246576406 %__expr_7843526591494960327))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_4799320818084596281 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-1459297673498492435) %__expr_-8637348939342624324)))) %__expr_2307122782736907488)) (NDArray___getitem__ %__expr_-4744074158064860754 (IndexKey_slice %__expr_-1459297673498492435)))) %__expr_6185990579165616281))) 0) +(TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none))))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_7773480733401006441)) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) (OptionalInt_none))))) 0) -(Int___eq__ (Int___mul__ %__expr_7773480733401006441 %__expr_-9127408350583128512) %__expr_-7750395295943998052) +(extract (Int_i64 (TupleInt_length (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))))) 0) +(Int___eq__ (Int___mul__ %__expr_3300329792522185471 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)) %__expr_2305922367412063525) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (Int___mul__ %__expr_7773480733401006441 %__expr_-9127408350583128512) %__expr_-7750395295943998052) 0) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) +(extract (Boolean_bool (Int___eq__ (Int___mul__ %__expr_3300329792522185471 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (OptionalDType_none) (OptionalBool_none) %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)) %__expr_2305922367412063525)) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) 0) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0))) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) 0) -(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1))) 0) +(Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3)) 0) -(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) +(extract (Boolean_bool (Int___ge__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 3))) 0) +(isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL)))) 0) -(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) +(extract (Boolean_bool (isdtype (NDArray_dtype (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (IsDtypeKind___or__ (IsDtypeKind_string "real floating") (IsDtypeKind___or__ (IsDtypeKind_string "complex floating") (IsDtypeKind_NULL))))) 0) +(Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none))))) 0) -(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) +(extract (Boolean_bool (Value_to_bool (NDArray_to_value (isfinite (sum (asarray (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)) (OptionalIntOrTuple_none)))))) 0) +(TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none)))) 0) -(Int___lt__ %__expr_-8392276577365791252 %__expr_7773480733401006441) +(extract (Int_i64 (TupleInt_length (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) %__expr_3300329792522185471) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ %__expr_-8392276577365791252 %__expr_7773480733401006441) 0) -(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) +(extract (Boolean_bool (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 0)) %__expr_3300329792522185471)) 0) +(Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2)) 0) -(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) +(extract (Boolean_bool (Int___eq__ (NDArray_ndim (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 2))) 0) +(Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1)) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441) %__expr_7773480733401006441) 0) -(TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) +(extract (Boolean_bool (Int___lt__ (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)) (Int___init__ 1))) 0) +(TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) (run-schedule (saturate (run array_api_ruleset))) -(extract (TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))))) 0) -(let %__expr_-2157517542993472809 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) -(Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_7773480733401006441) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441)) +(extract (Int_i64 (TupleInt_length (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))))) 0) +(let %__expr_-7766708656681124433 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) +(Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_3300329792522185471) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1))) (run-schedule (saturate (run array_api_ruleset))) -(extract (Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_7773480733401006441) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_-458881301673629543) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) %__expr_7773480733401006441)) 0) +(extract (Boolean_bool (Int___eq__ (TupleInt___getitem__ (NDArray_shape (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4)))))) %__expr_3300329792522185471) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") %__expr_2565664028803046192) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (OptionalDType_none) (OptionalBool_none) (OptionalDevice_none))) (Int___init__ 1)))) 0) (pop 1) (ruleset array_api_ruleset) (sort Int) @@ -704,22 +687,6 @@ (rewrite (f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__ f self i) (unstable-app f (TupleInt___getitem__ self i)) :ruleset array_api_ruleset) (function TupleInt_map (TupleInt UnstableFn_Int_Int) TupleInt :cost 100) (rewrite (TupleInt_map self f) (TupleInt___init__ (TupleInt_length self) (unstable-fn "f-UnstableFn_Int_Int_self-TupleInt_i-Int__unstable-app_f__TupleInt___getitem___self_i__" f self)) :ruleset array_api_ruleset) -(function i-TupleInt__-Int_i (TupleInt Int) TupleInt) -(rewrite (i-TupleInt__-Int_i i _) i :ruleset array_api_ruleset) -(sort TupleTupleInt) -(function TupleTupleInt_single (TupleInt) TupleTupleInt) -(sort UnstableFn_TupleInt_Int (UnstableFn (Int) TupleInt)) -(function TupleTupleInt___init__ (Int UnstableFn_TupleInt_Int) TupleTupleInt) -(rewrite (TupleTupleInt_single i) (TupleTupleInt___init__ (Int___init__ 1) (unstable-fn "i-TupleInt__-Int_i" i)) :ruleset array_api_ruleset) -(function TupleTupleInt_from_vec (Vec_Int) TupleInt) -(rewrite (TupleTupleInt_from_vec vec) (TupleInt___init__ (Int___init__ (vec-length vec)) (unstable-fn "index_vec_int" vec)) :ruleset array_api_ruleset) -(function TupleInt_if_ (Boolean TupleInt TupleInt) TupleInt) -(function TupleTupleInt_length (TupleTupleInt) Int) -(function TupleTupleInt___getitem__ (TupleTupleInt Int) TupleInt) -(function other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ (TupleTupleInt TupleTupleInt Int) TupleInt) -(rewrite (other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____ other self i) (TupleInt_if_ (Int___lt__ i (TupleTupleInt_length self)) (TupleTupleInt___getitem__ self i) (TupleTupleInt___getitem__ other (Int___sub__ i (TupleTupleInt_length self)))) :ruleset array_api_ruleset) -(function TupleTupleInt___add__ (TupleTupleInt TupleTupleInt) TupleTupleInt) -(rewrite (TupleTupleInt___add__ self other) (TupleTupleInt___init__ (Int___add__ (TupleTupleInt_length self) (TupleTupleInt_length other)) (unstable-fn "other-TupleTupleInt_self-TupleTupleInt_i-Int__TupleInt_if___Int___lt___i__TupleTupleInt_length_self____TupleTupleInt___getitem___self_i___TupleTupleInt___getitem___other__Int___sub___i__TupleTupleInt_length_self____" other self)) :ruleset array_api_ruleset) (sort NDArray) (function NDArray_size (NDArray) Int) (sort UnstableFn_Int_Int_Int (UnstableFn (Int Int) Int)) @@ -991,6 +958,7 @@ (sort UnstableFn_Boolean_Int (UnstableFn (Int) Boolean)) (function TupleInt_filter (TupleInt UnstableFn_Boolean_Int) TupleInt :cost 100) (rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ 0) idx_fn) filter_f) (TupleInt___init__ (Int___init__ 0) idx_fn) :ruleset array_api_ruleset) +(function TupleInt_if_ (Boolean TupleInt TupleInt) TupleInt) (rewrite (TupleInt_filter (TupleInt___init__ (Int___init__ k) idx_fn) filter_f) (TupleInt_if_ (unstable-app filter_f (unstable-app idx_fn (Int___init__ k))) (TupleInt___add__ (TupleInt_single (unstable-app idx_fn (Int___init__ k))) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) (TupleInt_filter (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)) filter_f)) :when ((!= k 0)) :ruleset array_api_ruleset) (function bottom_indexing (Int) Int) (rewrite (TupleInt_EMPTY) (TupleInt___init__ (Int___init__ 0) (unstable-fn "bottom_indexing")) :ruleset array_api_ruleset) @@ -1090,13 +1058,13 @@ (function Boolean___invert__ (Boolean) Boolean) (rewrite (Boolean___invert__ (TRUE)) (FALSE) :ruleset array_api_ruleset) (rewrite (Boolean___invert__ (FALSE)) (TRUE) :ruleset array_api_ruleset) -(ruleset ruleset_4998547360) +(ruleset ruleset_5033605696) (function NDArray___eq__ (NDArray NDArray) NDArray) -(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 1)) (NDArray_scalar (Value_int i))) (NDArray___eq__ x (NDArray_scalar (NDArray_index (unique_values x) (TupleInt_from_vec (vec-of i))))) :subsume :ruleset ruleset_4998547360) +(rewrite (NDArray___eq__ (TupleNDArray___getitem__ (unique_inverse x) (Int___init__ 1)) (NDArray_scalar (Value_int i))) (NDArray___eq__ x (NDArray_scalar (NDArray_index (unique_values x) (TupleInt_from_vec (vec-of i))))) :subsume :ruleset ruleset_5033605696) (function count_values (NDArray NDArray) TupleValue :unextractable) -(rewrite (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (NDArray_vector (count_values x (unique_values x))) :subsume :ruleset ruleset_4998547360) -(rewrite (count_values x (NDArray_vector (TupleValue___add__ (TupleValue___init__ v) tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) (count_values x (NDArray_vector tv))) :subsume :ruleset ruleset_4998547360) -(rewrite (count_values x (NDArray_vector (TupleValue___init__ v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_4998547360) +(rewrite (TupleNDArray___getitem__ (unique_counts x) (Int___init__ 1)) (NDArray_vector (count_values x (unique_values x))) :subsume :ruleset ruleset_5033605696) +(rewrite (count_values x (NDArray_vector (TupleValue___add__ (TupleValue___init__ v) tv))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) (count_values x (NDArray_vector tv))) :subsume :ruleset ruleset_5033605696) +(rewrite (count_values x (NDArray_vector (TupleValue___init__ v))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ x (NDArray_scalar v)) (OptionalIntOrTuple_none)))) :subsume :ruleset ruleset_5033605696) (function std (NDArray OptionalIntOrTuple) NDArray) (sort IntOrTuple) (function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) @@ -1104,28 +1072,23 @@ (function ndarray-sqrt (NDArray) NDArray) (function mean (NDArray OptionalIntOrTuple Boolean) NDArray) (function square (NDArray) NDArray) -(rewrite (std x (OptionalIntOrTuple_some (IntOrTuple_int i))) (ndarray-sqrt (mean (square (NDArray___sub__ x (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE))) :subsume :ruleset ruleset_4998547360) -(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) :subsume :ruleset ruleset_4998547360) +(rewrite (std x (OptionalIntOrTuple_some (IntOrTuple_int i))) (ndarray-sqrt (mean (square (NDArray___sub__ x (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)))) (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE))) :subsume :ruleset ruleset_5033605696) +(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (FALSE)) (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) :subsume :ruleset ruleset_5033605696) (function expand_dims (NDArray Int) NDArray) -(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) i) :subsume :ruleset ruleset_4998547360) -(let %__expr_4540085663845350143 (OptionalBool_none)) -(let %__expr_7530421370362517486 (OptionalDType_none)) -(let %__expr_-4082425080216461620 (OptionalDevice_none)) +(rewrite (mean x (OptionalIntOrTuple_some (IntOrTuple_int i)) (TRUE)) (expand_dims (NDArray___truediv__ (sum x (OptionalIntOrTuple_some (IntOrTuple_int i))) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape x) i)))) i) :subsume :ruleset ruleset_5033605696) +(let %__expr_-4122594143565222517 (OptionalBool_none)) +(let %__expr_-3674731718720096444 (OptionalDevice_none)) +(let %__expr_2091604061025500396 (OptionalDType_none)) (function NDArray_var (String) NDArray :cost 200) -(let %__expr_9189154200368037574 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_4540085663845350143) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) -(let %__expr_7773480733401006441 (Int___init__ 1)) -(let %__expr_1906307530420321623 (IndexKey_int (Int___init__ 0))) -(let %__expr_871583677900651896 (Int___init__ 0)) -(let %__expr_4873895522838289209 (OptionalInt_none)) -(let %__expr_6308258715381497904 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray %__expr_9189154200368037574 %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) %__expr_4873895522838289209))) -(let %__expr_-7750395295943998052 (Int___init__ 2)) -(let %__expr_-8392276577365791252 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_871583677900651896)) -(let %__expr_-9127408350583128512 (TupleInt___getitem__ (NDArray_shape %__expr_6308258715381497904) %__expr_871583677900651896)) -(let %__expr_-8852798553151442138 (Value_int (Int___init__ 2))) -(let %__expr_8044523218021318818 (Value_int (Int___init__ 1))) -(let %__expr_3713283541138310173 (Int___init__ 150)) -(let %__expr_7039565710158856600 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) -(let %__expr_-2697243261424336135 (Value_int (Int___init__ 0))) +(let %__expr_-5200143055942215294 (asarray (reshape (asarray (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444) (TupleInt_from_vec (vec-of (Int___init__ -1))) %__expr_-4122594143565222517) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) +(let %__expr_3300329792522185471 (Int___init__ 1)) +(let %__expr_-7942988945600443502 (Int___init__ 0)) +(let %__expr_2407075622821899710 (TupleInt___getitem__ (NDArray_shape (unique_values (concat (TupleNDArray___init__ (unique_values (asarray %__expr_-5200143055942215294 %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) %__expr_-7942988945600443502)) +(let %__expr_-143818766793636568 (OptionalInt_none)) +(let %__expr_2305922367412063525 (Int___init__ 2)) +(let %__expr_-1113027338580586528 (unique_values (concat (TupleNDArray___init__ (unique_values (asarray %__expr_-5200143055942215294 %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (OptionalInt_none)))) +(let %__expr_4326809257670680665 (IndexKey_int %__expr_-7942988945600443502)) +(let %__expr_-6932009612334166206 (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_-7942988945600443502)) (sort MultiAxisIndexKey) (function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) (sort MultiAxisIndexKeyItem) @@ -1135,10 +1098,18 @@ (sort Slice) (function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) (function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) -(let %__expr_811525261394403186 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7750395295943998052) (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 %__expr_4873895522838289209 %__expr_4873895522838289209)))))) -(let %__expr_4396021293866471344 (TupleNDArray___getitem__ (unique_inverse %__expr_9189154200368037574) %__expr_7773480733401006441)) -(let %__expr_-7672334842805227 (unique_inverse %__expr_9189154200368037574)) -(let %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 %__expr_4873895522838289209 %__expr_4873895522838289209))) +(let %__expr_-954298076671303615 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_2305922367412063525) (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 %__expr_-143818766793636568 %__expr_-143818766793636568)))))) +(let %__expr_483064111004974365 (Float___init__ 1.0)) +(let %__expr_-3520630818915629594 (Int___init__ 150)) +(let %__expr_-6765600943420531488 (Value_int (Int___init__ 2))) +(let %__expr_-622493510873225930 (Value_int (Int___init__ 1))) +(let %__expr_478758938070675600 (Value_int (Int___init__ 0))) +(let %__expr_-8664633617851008504 (unique_inverse %__expr_-5200143055942215294)) +(let %__expr_-8613124149453883591 (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) +(let %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 %__expr_-143818766793636568 %__expr_-143818766793636568))) +(let %__expr_-2704792733181451377 (TupleNDArray___getitem__ %__expr_-8664633617851008504 %__expr_3300329792522185471)) +(let %__expr_1913846200346632608 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_-7942988945600443502))) +(let %__expr_6185990579165616281 (FALSE)) (function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) (function zeros (TupleInt OptionalDType OptionalDevice) NDArray) (function OptionalDType_some (DType) OptionalDType) @@ -1146,400 +1117,257 @@ (function OptionalDevice_some (Device) OptionalDevice) (function NDArray_device (NDArray) Device) (function IndexKey_ndarray (NDArray) IndexKey) -(let %__expr_4481754591798653144 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ %__expr_-7672334842805227 %__expr_871583677900651896)) %__expr_871583677900651896) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) %__expr_7773480733401006441))) (OptionalDType_some (NDArray_dtype %__expr_7039565710158856600)) (OptionalDevice_some (NDArray_device %__expr_7039565710158856600))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_-2697243261424336135)))) (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896)) (FALSE))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569))) (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_8044523218021318818)))) (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896)) (FALSE))) %__expr_811525261394403186 (mean (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_4396021293866471344 (NDArray_scalar %__expr_-8852798553151442138)))) (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896)) (FALSE)))) -(let %__expr_-907407888652541496 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569)))) -(let %__expr_-6354454737791578173 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))) -(let %__expr_4948775059192706173 (NDArray_scalar %__expr_-2697243261424336135)) -(let %__expr_6624227409259184391 (Float___init__ 1.0)) -(let %__expr_4942768151788412671 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620))) -(let %__expr_-6383682287120045686 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896))) -(let %__expr_5801044028809884995 (FALSE)) +(let %__expr_7843526591494960327 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (TupleInt___getitem__ (NDArray_shape (TupleNDArray___getitem__ %__expr_-8664633617851008504 %__expr_-7942988945600443502)) %__expr_-7942988945600443502) (TupleInt___getitem__ (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) %__expr_3300329792522185471))) (OptionalDType_some (NDArray_dtype %__expr_-8613124149453883591)) (OptionalDevice_some (NDArray_device %__expr_-8613124149453883591))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7942988945600443502) %__expr_-8637348939342624324))) (mean (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-2704792733181451377 (NDArray_scalar %__expr_478758938070675600)))) %__expr_1913846200346632608 %__expr_6185990579165616281)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_3300329792522185471) %__expr_-8637348939342624324))) (mean (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-2704792733181451377 (NDArray_scalar %__expr_-622493510873225930)))) %__expr_1913846200346632608 %__expr_6185990579165616281)) %__expr_-954298076671303615 (mean (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-2704792733181451377 (NDArray_scalar %__expr_-6765600943420531488)))) %__expr_1913846200346632608 %__expr_6185990579165616281))) +(let %__expr_7180621784385959982 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_3300329792522185471) %__expr_-8637348939342624324)))) +(let %__expr_-3283806677371465170 (NDArray_scalar %__expr_478758938070675600)) +(let %__expr_-4323225966908438009 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7942988945600443502) %__expr_-8637348939342624324)))) +(let %__expr_-8475266842302933511 (NDArray_shape (asarray (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444))) (function OptionalInt_some (Int) OptionalInt) -(let %__expr_-7185048337851367263 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 %__expr_1906307530420321623)))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-907407888652541496))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_7773480733401006441))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_-6354454737791578173))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_7039565710158856600 (IndexKey_ndarray (NDArray___eq__ %__expr_9189154200368037574 (NDArray___getitem__ %__expr_6308258715381497904 (IndexKey_int %__expr_-7750395295943998052))))) (NDArray___getitem__ %__expr_4481754591798653144 %__expr_811525261394403186))))) (OptionalInt_some %__expr_871583677900651896))) +(let %__expr_3049932803880609425 (std (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 %__expr_4326809257670680665)))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-4323225966908438009))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_3300329792522185471))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_7180621784385959982))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_2305922367412063525))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-954298076671303615))))) (OptionalInt_some %__expr_-7942988945600443502)) %__expr_1913846200346632608)) +(let %__expr_-671931098406187877 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 %__expr_4326809257670680665)))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-4323225966908438009))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_3300329792522185471))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_7180621784385959982))) (TupleNDArray___init__ (NDArray___sub__ (NDArray___getitem__ %__expr_-8613124149453883591 (IndexKey_ndarray (NDArray___eq__ %__expr_-5200143055942215294 (NDArray___getitem__ %__expr_-1113027338580586528 (IndexKey_int %__expr_2305922367412063525))))) (NDArray___getitem__ %__expr_7843526591494960327 %__expr_-954298076671303615))))) (OptionalInt_some %__expr_-7942988945600443502))) (function NDArray___mul__ (NDArray NDArray) NDArray) -(let %__expr_1166113493069915661 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-8392276577365791252 %__expr_-9127408350583128512))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995) %__expr_7773480733401006441)) -(let %__expr_-2809124111957918512 (std %__expr_-7185048337851367263 %__expr_-6383682287120045686)) -(let %__expr_-1704542949091505620 (NDArray_dtype %__expr_7039565710158856600)) -(let %__expr_-2471947197794367259 (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_1166113493069915661 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_4873895522838289209)) -(let %__expr_5778560295018625428 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-8392276577365791252 %__expr_-9127408350583128512))))) %__expr_7530421370362517486 %__expr_4540085663845350143 %__expr_-4082425080216461620)) (NDArray___truediv__ %__expr_-7185048337851367263 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_5801044028809884995)) +(let %__expr_-1459297673498492435 (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ %__expr_-6932009612334166206 %__expr_2407075622821899710))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281) %__expr_3300329792522185471) (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) (OptionalIntOrTuple_none))))) %__expr_-143818766793636568)) +(let %__expr_4799320818084596281 (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ %__expr_-6932009612334166206 %__expr_2407075622821899710))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281)) +(let %__expr_-4744074158064860754 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (asarray (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ %__expr_-6932009612334166206 %__expr_2407075622821899710))))) %__expr_2091604061025500396 %__expr_-4122594143565222517 %__expr_-3674731718720096444)) (NDArray___truediv__ %__expr_-671931098406187877 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) %__expr_6185990579165616281) %__expr_3300329792522185471)) +(let %__expr_2307122782736907488 (NDArray___setitem__ %__expr_3049932803880609425 (IndexKey_ndarray (NDArray___eq__ %__expr_3049932803880609425 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) (function NDArray___matmul__ (NDArray NDArray) NDArray) (function IndexKey_slice (Slice) IndexKey) -(let %__expr_2843694094811160801 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-9127408350583128512 %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995) %__expr_7773480733401006441)) -(let %__expr_-1190917143489565870 (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0))))) -(let %__expr_3952577209981612471 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-8392276577365791252)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_6624227409259184391 (Float_from_int (Int___sub__ %__expr_-9127408350583128512 %__expr_7773480733401006441))))))) (NDArray_T (NDArray___sub__ %__expr_4481754591798653144 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) %__expr_5801044028809884995)) -(let %__expr_-3429845910624471226 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) -(let %__expr_7346177422650294070 (DType_int32)) -(let %__expr_-7476478225777086989 (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0))))) -(let %__expr_-8563557304220193917 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_9189154200368037574) %__expr_7773480733401006441) %__expr_-1704542949091505620) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_4481754591798653144)) -(let %__expr_-5852474775256756550 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_5778560295018625428 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-2471947197794367259) %__expr_4574319845551051569)))) (NDArray___setitem__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) (IndexKey_ndarray (NDArray___eq__ (std %__expr_-7185048337851367263 %__expr_-6383682287120045686) %__expr_4948775059192706173)) (NDArray_scalar (Value_float (Float___init__ 1.0)))))) (NDArray___getitem__ %__expr_1166113493069915661 (IndexKey_slice %__expr_-2471947197794367259)))) -(simplify (saturate (seq (run array_api_ruleset) (run ruleset_4998547360))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_7039565710158856600 %__expr_-8563557304220193917) (NDArray___matmul__ %__expr_-5852474775256756550 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_3952577209981612471 %__expr_-7750395295943998052)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_2843694094811160801 (NDArray___mul__ %__expr_-3429845910624471226 (NDArray___getitem__ %__expr_2843694094811160801 %__expr_1906307530420321623))) %__expr_7346177422650294070) (OptionalIntOrTuple_none))))) %__expr_4873895522838289209)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some %__expr_-7750395295943998052) %__expr_4873895522838289209))))))) -(pop 1) -(sort Int) -(function Int___init__ (i64) Int) -(let %__expr_871583677900651896 (Int___init__ 0)) -(sort Value) -(function Value_int (Int) Value) -(let %__expr_8044523218021318818 (Value_int (Int___init__ 1))) -(let %__expr_7773480733401006441 (Int___init__ 1)) -(let %__expr_3188239554967046364 (Int___init__ 4)) -(let %__expr_3713283541138310173 (Int___init__ 150)) -(sort NDArray) -(sort IndexKey) -(function NDArray___getitem__ (NDArray IndexKey) NDArray) -(function assume_isfinite (NDArray) NDArray) -(sort TupleInt) -(function assume_shape (NDArray TupleInt) NDArray) -(sort DType) -(function assume_dtype (NDArray DType) NDArray) -(function NDArray_var (String) NDArray :cost 200) -(function DType_float64 () DType) -(sort Vec_Int (Vec Int)) -(function TupleInt_from_vec (Vec_Int) TupleInt) -(function IndexKey_ndarray (NDArray) IndexKey) -(function NDArray___eq__ (NDArray NDArray) NDArray) -(sort TupleValue) -(function assume_value_one_of (NDArray TupleValue) NDArray) -(function DType_int64 () DType) -(function TupleValue___add__ (TupleValue TupleValue) TupleValue) -(function TupleValue___init__ (Value) TupleValue) -(function NDArray_scalar (Value) NDArray) -(let %__expr_-8193044930268358572 (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173 %__expr_3188239554967046364)))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_8044523218021318818))))) -(let %__expr_-8852798553151442138 (Value_int (Int___init__ 2))) -(let %__expr_7894034366188933940 (NDArray___getitem__ (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173 %__expr_3188239554967046364)))) (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-8852798553151442138))))) -(sort MultiAxisIndexKeyItem) -(sort Slice) -(function MultiAxisIndexKeyItem_slice (Slice) MultiAxisIndexKeyItem) -(sort OptionalInt) -(function Slice___init__ (OptionalInt OptionalInt OptionalInt) Slice) -(function OptionalInt_none () OptionalInt) -(let %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ (OptionalInt_none) (OptionalInt_none) (OptionalInt_none)))) -(let %__expr_-8148879634944358227 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173 %__expr_3188239554967046364))))) -(let %__expr_-2697243261424336135 (Value_int %__expr_871583677900651896)) -(let %__expr_-6996286770058887658 (NDArray___getitem__ %__expr_-8148879634944358227 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-2697243261424336135))))) -(let %__expr_-6000263953540950166 (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2))))))) -(let %__expr_-7750395295943998052 (Int___init__ 2)) -(sort MultiAxisIndexKey) -(function IndexKey_multi_axis (MultiAxisIndexKey) IndexKey) -(sort Vec_MultiAxisIndexKeyItem (Vec MultiAxisIndexKeyItem)) -(function MultiAxisIndexKey_from_vec (Vec_MultiAxisIndexKeyItem) MultiAxisIndexKey) -(function MultiAxisIndexKeyItem_int (Int) MultiAxisIndexKeyItem) -(let %__expr_852382103983847011 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7750395295943998052) %__expr_4574319845551051569)))) -(let %__expr_-458881301673629543 (DType_float64)) -(sort OptionalIntOrTuple) -(sort IntOrTuple) -(function OptionalIntOrTuple_some (IntOrTuple) OptionalIntOrTuple) -(function IntOrTuple_int (Int) IntOrTuple) -(let %__expr_-6383682287120045686 (OptionalIntOrTuple_some (IntOrTuple_int %__expr_871583677900651896))) -(function NDArray___setitem__ (NDArray IndexKey NDArray) NDArray) -(sort OptionalDType) -(sort OptionalDevice) -(function zeros (TupleInt OptionalDType OptionalDevice) NDArray) -(function OptionalDType_some (DType) OptionalDType) -(sort Device) -(function OptionalDevice_some (Device) OptionalDevice) -(function NDArray_device (NDArray) Device) -(function NDArray___truediv__ (NDArray NDArray) NDArray) -(function sum (NDArray OptionalIntOrTuple) NDArray) -(function TupleInt___getitem__ (TupleInt Int) Int) -(function NDArray_shape (NDArray) TupleInt) -(let %__expr_1269092575758336087 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) %__expr_3188239554967046364)) (OptionalDType_some %__expr_-458881301673629543) (OptionalDevice_some (NDArray_device %__expr_-8148879634944358227))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569))) (NDArray___truediv__ (sum %__expr_-6996286770058887658 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-6996286770058887658) %__expr_871583677900651896))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569))) (NDArray___truediv__ (sum %__expr_-8193044930268358572 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-8193044930268358572) %__expr_871583677900651896))))) %__expr_852382103983847011 (NDArray___truediv__ (sum %__expr_7894034366188933940 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_7894034366188933940) %__expr_871583677900651896)))))) -(let %__expr_637321235555326848 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_871583677900651896) %__expr_4574319845551051569)))) -(sort TupleNDArray) -(function concat (TupleNDArray OptionalInt) NDArray) -(function TupleNDArray___add__ (TupleNDArray TupleNDArray) TupleNDArray) -(function TupleNDArray___init__ (NDArray) TupleNDArray) -(function NDArray___sub__ (NDArray NDArray) NDArray) -(function OptionalInt_some (Int) OptionalInt) -(let %__expr_3786928033384172571 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-6996286770058887658 (NDArray___getitem__ %__expr_1269092575758336087 %__expr_637321235555326848))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-8193044930268358572 (NDArray___getitem__ %__expr_1269092575758336087 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))))) (TupleNDArray___init__ (NDArray___sub__ %__expr_7894034366188933940 (NDArray___getitem__ %__expr_1269092575758336087 %__expr_852382103983847011))))) (OptionalInt_some %__expr_871583677900651896))) -(function OptionalIntOrTuple_none () OptionalIntOrTuple) -(let %__expr_8391745089977789030 (OptionalIntOrTuple_none)) -(let %__expr_6937128598082130908 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-8852798553151442138))) -(let %__expr_-7260687232290637626 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_8044523218021318818))) -(let %__expr_-167787087406110176 1) -(function square (NDArray) NDArray) -(function expand_dims (NDArray Int) NDArray) -(let %__expr_8477700738181691910 (square (NDArray___sub__ %__expr_3786928033384172571 (expand_dims (NDArray___truediv__ (sum %__expr_3786928033384172571 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_3786928033384172571) %__expr_871583677900651896)))) %__expr_871583677900651896)))) -(let %__expr_8988307273465367651 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_7773480733401006441) %__expr_4574319845551051569)))) -(let %__expr_4857534354917498587 2) -(function ndarray-sqrt (NDArray) NDArray) -(let %__expr_3277279022187426799 (ndarray-sqrt (NDArray___truediv__ (sum %__expr_8477700738181691910 %__expr_-6383682287120045686) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_8477700738181691910) %__expr_871583677900651896)))))) -(function TupleNDArray___getitem__ (TupleNDArray Int) NDArray) -(sort Boolean) -(function svd (NDArray Boolean) TupleNDArray) -(function NDArray___mul__ (NDArray NDArray) NDArray) -(sort Float) -(function Value_float (Float) Value) -(function Float_rational (Rational) Float :cost 2) -(function FALSE () Boolean) -(let %__expr_-6992665957909292539 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 147))))) (NDArray___truediv__ %__expr_3786928033384172571 (NDArray___setitem__ %__expr_3277279022187426799 (IndexKey_ndarray (NDArray___eq__ %__expr_3277279022187426799 (NDArray_scalar %__expr_-2697243261424336135))) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_-167787087406110176))))))) (FALSE)) %__expr_7773480733401006441)) -(let %__expr_4948775059192706173 (NDArray_scalar %__expr_-2697243261424336135)) -(let %__expr_4873895522838289209 (OptionalInt_none)) -(function Value_to_int (Value) Int) -(function NDArray_to_value (NDArray) Value) -(function astype (NDArray DType) NDArray) -(function NDArray___gt__ (NDArray NDArray) NDArray) -(function Float___init__ (f64) Float :cost 3) -(function DType_int32 () DType) -(let %__expr_5508951548654840646 (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_-6992665957909292539 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) (DType_int32)) %__expr_8391745089977789030)))) %__expr_4873895522838289209)) -(let %__expr_-2951635939159740441 150) -(function NDArray_vector (TupleValue) NDArray) -(let %__expr_-3755709727314850279 (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-2697243261424336135)) %__expr_8391745089977789030))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-7260687232290637626 %__expr_8391745089977789030))) (TupleValue___init__ (NDArray_to_value (sum %__expr_6937128598082130908 %__expr_8391745089977789030)))))) %__expr_-458881301673629543) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-2951635939159740441 %__expr_-167787087406110176)))))) -(let %__expr_5561760508855517538 (NDArray___setitem__ %__expr_3277279022187426799 (IndexKey_ndarray (NDArray___eq__ %__expr_3277279022187426799 (NDArray_scalar %__expr_-2697243261424336135))) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_-167787087406110176)))))) -(let %__expr_8075847335762488630 (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 147))))) (NDArray___truediv__ %__expr_3786928033384172571 (NDArray___setitem__ %__expr_3277279022187426799 (IndexKey_ndarray (NDArray___eq__ %__expr_3277279022187426799 (NDArray_scalar %__expr_-2697243261424336135))) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_-167787087406110176))))))) (FALSE))) -(let %__expr_5801044028809884995 (FALSE)) -(function NDArray___matmul__ (NDArray NDArray) NDArray) -(function NDArray_T (NDArray) NDArray) -(function IndexKey_slice (Slice) IndexKey) -(let %__expr_-7611578709264289821 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_3713283541138310173)) %__expr_-3755709727314850279) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-167787087406110176 %__expr_4857534354917498587)))))) (NDArray_T (NDArray___sub__ %__expr_1269092575758336087 (NDArray___matmul__ %__expr_-3755709727314850279 %__expr_1269092575758336087))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_8075847335762488630 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_5508951548654840646) %__expr_4574319845551051569)))) %__expr_5561760508855517538)) (NDArray___getitem__ %__expr_-6992665957909292539 (IndexKey_slice %__expr_5508951548654840646)))) %__expr_5801044028809884995)) -(let %__expr_-3429845910624471226 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) -(let %__expr_7346177422650294070 (DType_int32)) -(let %__expr_-2626596927381929382 (NDArray_var "y")) -(let %__expr_-8462033748461398887 (NDArray___matmul__ %__expr_-3755709727314850279 %__expr_1269092575758336087)) -(let %__expr_3830746366128651238 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_3713283541138310173))) (TupleValue___add__ (TupleValue___init__ (Value_int %__expr_871583677900651896)) (TupleValue___add__ (TupleValue___init__ %__expr_8044523218021318818) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-2697243261424336135))) -(let %__expr_-1892903834471356524 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_8075847335762488630 %__expr_-7750395295943998052) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_5508951548654840646) %__expr_4574319845551051569)))) %__expr_5561760508855517538)) (NDArray___getitem__ %__expr_-6992665957909292539 (IndexKey_slice %__expr_5508951548654840646)))) -(let %__expr_5036000968216100388 (NDArray_var "X")) +(let %__expr_270530672567316052 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-6932009612334166206)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-5200143055942215294) %__expr_3300329792522185471) (NDArray_dtype %__expr_-8613124149453883591)) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ %__expr_2407075622821899710 %__expr_3300329792522185471))))))) (NDArray_T (NDArray___sub__ %__expr_7843526591494960327 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-5200143055942215294) %__expr_3300329792522185471) (NDArray_dtype %__expr_-8613124149453883591)) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_7843526591494960327))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_4799320818084596281 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-1459297673498492435) %__expr_-8637348939342624324)))) %__expr_2307122782736907488)) (NDArray___getitem__ %__expr_-4744074158064860754 (IndexKey_slice %__expr_-1459297673498492435)))) %__expr_6185990579165616281) %__expr_3300329792522185471)) +(let %__expr_-5540247018246576406 (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-5200143055942215294) %__expr_3300329792522185471) (NDArray_dtype %__expr_-8613124149453883591)) (NDArray_scalar (Value_float (Float___init__ 150.0))))) +(let %__expr_-5933626871507061239 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-6932009612334166206)) (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-5200143055942215294) %__expr_3300329792522185471) (NDArray_dtype %__expr_-8613124149453883591)) (NDArray_scalar (Value_float (Float___init__ 150.0))))) (NDArray_scalar (Value_float (Float___truediv__ %__expr_483064111004974365 (Float_from_int (Int___sub__ %__expr_2407075622821899710 %__expr_3300329792522185471))))))) (NDArray_T (NDArray___sub__ %__expr_7843526591494960327 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-5200143055942215294) %__expr_3300329792522185471) (NDArray_dtype %__expr_-8613124149453883591)) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_7843526591494960327))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_4799320818084596281 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-1459297673498492435) %__expr_-8637348939342624324)))) %__expr_2307122782736907488)) (NDArray___getitem__ %__expr_-4744074158064860754 (IndexKey_slice %__expr_-1459297673498492435)))) %__expr_6185990579165616281)) +(let %__expr_-4180228001421012884 (NDArray_scalar (Value_float (Float___init__ 0.0001)))) +(let %__expr_-3501279694271260104 (DType_int32)) +(let %__expr_5981948135265815040 (OptionalIntOrTuple_none)) +(let %__expr_2917853313867759590 (NDArray_dtype %__expr_-8613124149453883591)) +(let %__expr_-5161099022452562755 (NDArray___matmul__ (NDArray___truediv__ (astype (TupleNDArray___getitem__ (unique_counts %__expr_-5200143055942215294) %__expr_3300329792522185471) (NDArray_dtype %__expr_-8613124149453883591)) (NDArray_scalar (Value_float (Float___init__ 150.0)))) %__expr_7843526591494960327)) +(let %__expr_7947521701427986423 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_4799320818084596281 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_-1459297673498492435) %__expr_-8637348939342624324)))) %__expr_2307122782736907488)) (NDArray___getitem__ %__expr_-4744074158064860754 (IndexKey_slice %__expr_-1459297673498492435)))) +(simplify (saturate (seq (run array_api_ruleset) (run ruleset_5033605696))) (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-8613124149453883591 %__expr_-5161099022452562755) (NDArray___matmul__ %__expr_7947521701427986423 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-5933626871507061239 %__expr_2305922367412063525)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_270530672567316052 (NDArray___mul__ %__expr_-4180228001421012884 (NDArray___getitem__ %__expr_270530672567316052 %__expr_4326809257670680665))) %__expr_-3501279694271260104) %__expr_5981948135265815040)))) %__expr_-143818766793636568)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some %__expr_2305922367412063525) %__expr_-143818766793636568))))))) +(let %__expr_978886785202701864 (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594 (Int___init__ 4)))))) +(let %__expr_3826936542596060759 (Int___init__ 4)) +(let %__expr_-8607724285922497289 (NDArray___getitem__ %__expr_978886785202701864 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-622493510873225930))))) +(let %__expr_8779256037532389381 (NDArray___getitem__ %__expr_978886785202701864 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_-3283806677371465170)))) +(let %__expr_2565664028803046192 (DType_float64)) +(let %__expr_3562727526135681919 (NDArray___getitem__ %__expr_978886785202701864 (IndexKey_ndarray (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-6765600943420531488))))) +(let %__expr_5127641217187406832 (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2))))))) +(let %__expr_-5874110021677832412 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_-7942988945600443502) %__expr_-8637348939342624324)))) +(let %__expr_2924140249920978521 (NDArray___setitem__ (NDArray___setitem__ (NDArray___setitem__ (zeros (TupleInt_from_vec (vec-of (Int___init__ 3) %__expr_3826936542596060759)) (OptionalDType_some %__expr_2565664028803046192) (OptionalDevice_some (NDArray_device %__expr_978886785202701864))) %__expr_-5874110021677832412 (NDArray___truediv__ (sum %__expr_8779256037532389381 %__expr_1913846200346632608) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_8779256037532389381) %__expr_-7942988945600443502))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_3300329792522185471) %__expr_-8637348939342624324))) (NDArray___truediv__ (sum %__expr_-8607724285922497289 %__expr_1913846200346632608) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-8607724285922497289) %__expr_-7942988945600443502))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_2305922367412063525) %__expr_-8637348939342624324))) (NDArray___truediv__ (sum %__expr_3562727526135681919 %__expr_1913846200346632608) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_3562727526135681919) %__expr_-7942988945600443502)))))) +(let %__expr_8582344284568866371 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_2305922367412063525) %__expr_-8637348939342624324)))) +(let %__expr_-3132706819403651174 (concat (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_8779256037532389381 (NDArray___getitem__ %__expr_2924140249920978521 %__expr_-5874110021677832412))) (TupleNDArray___add__ (TupleNDArray___init__ (NDArray___sub__ %__expr_-8607724285922497289 (NDArray___getitem__ %__expr_2924140249920978521 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_3300329792522185471) %__expr_-8637348939342624324)))))) (TupleNDArray___init__ (NDArray___sub__ %__expr_3562727526135681919 (NDArray___getitem__ %__expr_2924140249920978521 %__expr_8582344284568866371))))) (OptionalInt_some %__expr_-7942988945600443502))) +(let %__expr_755766186975356825 (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_int %__expr_3300329792522185471) %__expr_-8637348939342624324)))) +(let %__expr_1540266631570115279 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-622493510873225930))) +(let %__expr_7382830492167037887 (square (NDArray___sub__ %__expr_-3132706819403651174 (expand_dims (NDArray___truediv__ (sum %__expr_-3132706819403651174 %__expr_1913846200346632608) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_-3132706819403651174) %__expr_-7942988945600443502)))) %__expr_-7942988945600443502)))) +(let %__expr_-8085216922850384227 1) +(let %__expr_-6293504624159906607 (ndarray-sqrt (NDArray___truediv__ (sum %__expr_7382830492167037887 %__expr_1913846200346632608) (NDArray_scalar (Value_int (TupleInt___getitem__ (NDArray_shape %__expr_7382830492167037887) %__expr_-7942988945600443502)))))) +(let %__expr_6749312899897339687 (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 147))))) (NDArray___truediv__ %__expr_-3132706819403651174 (NDArray___setitem__ %__expr_-6293504624159906607 (IndexKey_ndarray (NDArray___eq__ %__expr_-6293504624159906607 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 %__expr_-8085216922850384227))))))) %__expr_6185990579165616281) %__expr_3300329792522185471) %__expr_-4180228001421012884) %__expr_-3501279694271260104) %__expr_5981948135265815040)))) %__expr_-143818766793636568)) +(let %__expr_-5507881356079575778 (TupleNDArray___getitem__ (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 147))))) (NDArray___truediv__ %__expr_-3132706819403651174 (NDArray___setitem__ %__expr_-6293504624159906607 (IndexKey_ndarray (NDArray___eq__ %__expr_-6293504624159906607 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 %__expr_-8085216922850384227))))))) %__expr_6185990579165616281) %__expr_3300329792522185471)) +(let %__expr_-649824781713772813 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) %__expr_-3283806677371465170)) +(let %__expr_-7715872950595615314 (svd (NDArray___mul__ (ndarray-sqrt (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 147))))) (NDArray___truediv__ %__expr_-3132706819403651174 (NDArray___setitem__ %__expr_-6293504624159906607 (IndexKey_ndarray (NDArray___eq__ %__expr_-6293504624159906607 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 %__expr_-8085216922850384227))))))) %__expr_6185990579165616281)) +(let %__expr_-3823036485969157667 150) +(let %__expr_-5209865760313326585 (NDArray___truediv__ (astype (NDArray_vector (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_-649824781713772813 %__expr_5981948135265815040))) (TupleValue___add__ (TupleValue___init__ (NDArray_to_value (sum %__expr_1540266631570115279 %__expr_5981948135265815040))) (TupleValue___init__ (NDArray_to_value (sum (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-6765600943420531488)) %__expr_5981948135265815040)))))) %__expr_2565664028803046192) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-3823036485969157667 %__expr_-8085216922850384227)))))) +(let %__expr_3134672938439589730 (NDArray___setitem__ %__expr_-6293504624159906607 (IndexKey_ndarray (NDArray___eq__ %__expr_-6293504624159906607 %__expr_-3283806677371465170)) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 %__expr_-8085216922850384227)))))) +(let %__expr_2615935926139764667 (TupleNDArray___getitem__ (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-3520630818915629594)) %__expr_-5209865760313326585) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 2)))))) (NDArray_T (NDArray___sub__ %__expr_2924140249920978521 (NDArray___matmul__ %__expr_-5209865760313326585 %__expr_2924140249920978521))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-7715872950595615314 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_6749312899897339687) %__expr_-8637348939342624324)))) %__expr_3134672938439589730)) (NDArray___getitem__ %__expr_-5507881356079575778 (IndexKey_slice %__expr_6749312899897339687)))) %__expr_6185990579165616281) %__expr_3300329792522185471)) +(let %__expr_-3749619018425312853 2) +(let %__expr_-4464433185064523998 (svd (NDArray___matmul__ (NDArray_T (NDArray___mul__ (ndarray-sqrt (NDArray___mul__ (NDArray___mul__ (NDArray_scalar (Value_int %__expr_-3520630818915629594)) %__expr_-5209865760313326585) (NDArray_scalar (Value_float (Float_rational (rational %__expr_-8085216922850384227 2)))))) (NDArray_T (NDArray___sub__ %__expr_2924140249920978521 (NDArray___matmul__ %__expr_-5209865760313326585 %__expr_2924140249920978521))))) (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-7715872950595615314 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_6749312899897339687) %__expr_-8637348939342624324)))) %__expr_3134672938439589730)) (NDArray___getitem__ %__expr_-5507881356079575778 (IndexKey_slice %__expr_6749312899897339687)))) %__expr_6185990579165616281)) +(let %__expr_8528844779137627870 (NDArray___eq__ (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of %__expr_-3520630818915629594))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) (NDArray_scalar %__expr_-6765600943420531488))) +(let %__expr_4447619498771141281 (NDArray___matmul__ %__expr_-5209865760313326585 %__expr_2924140249920978521)) +(let %__expr_-2753571644920605745 (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))) +(let %__expr_3259058749987290790 (assume_dtype (NDArray_var "y") (DType_int64))) +(let %__expr_-6562228218073754698 (NDArray___truediv__ (NDArray_T (NDArray___truediv__ (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-7715872950595615314 %__expr_2305922367412063525) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of (MultiAxisIndexKeyItem_slice %__expr_6749312899897339687) %__expr_-8637348939342624324)))) %__expr_3134672938439589730)) (NDArray___getitem__ %__expr_-5507881356079575778 (IndexKey_slice %__expr_6749312899897339687)))) +(let %__expr_-1580149342035447281 0) (sort Program) -(function ndarray_function_two (NDArray NDArray NDArray) Program) -(function IndexKey_int (Int) IndexKey) -(let %fn_program (ndarray_function_two (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_-8148879634944358227 %__expr_-8462033748461398887) (NDArray___matmul__ %__expr_-1892903834471356524 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-7611578709264289821 %__expr_-7750395295943998052)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ (TupleNDArray___getitem__ %__expr_-7611578709264289821 %__expr_7773480733401006441) (NDArray___mul__ %__expr_-3429845910624471226 (NDArray___getitem__ (TupleNDArray___getitem__ %__expr_-7611578709264289821 %__expr_7773480733401006441) (IndexKey_int %__expr_871583677900651896)))) %__expr_7346177422650294070) %__expr_8391745089977789030)))) %__expr_4873895522838289209)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_4574319845551051569 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_4873895522838289209 (OptionalInt_some %__expr_-7750395295943998052) %__expr_4873895522838289209)))))) %__expr_5036000968216100388 %__expr_-2626596927381929382)) -(ruleset ruleset_4998548992) +(function Program_compile (Program i64) Unit :default ()) +(function ndarray_function_two_program (NDArray NDArray NDArray) Program) +(Program_compile (ndarray_function_two_program (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_978886785202701864 %__expr_4447619498771141281) (NDArray___matmul__ %__expr_-6562228218073754698 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-4464433185064523998 %__expr_2305922367412063525)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_2615935926139764667 (NDArray___mul__ %__expr_-4180228001421012884 (NDArray___getitem__ %__expr_2615935926139764667 %__expr_4326809257670680665))) %__expr_-3501279694271260104) %__expr_5981948135265815040)))) %__expr_-143818766793636568)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some %__expr_2305922367412063525) %__expr_-143818766793636568)))))) (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2))))))) %__expr_-1580149342035447281) +(ruleset array_api_program_gen_ruleset) +(function Program_function_two (Program Program Program String) Program) (function ndarray_program (NDArray) Program) +(rewrite (ndarray_function_two_program res l r) (Program_function_two (ndarray_program res) (ndarray_program l) (ndarray_program r) "__fn") :ruleset array_api_program_gen_ruleset) (function Program___init__ (String bool) Program) -(rewrite (ndarray_program (NDArray_var s)) (Program___init__ s true) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray_var s)) (Program___init__ s true) :ruleset array_api_program_gen_ruleset) (function Program_statement (Program Program) Program :unextractable) (function Program___add__ (Program Program) Program) (function dtype_program (DType) Program) -(rewrite (ndarray_program (assume_dtype z dtype)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".dtype == " false)) (dtype_program dtype))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (assume_dtype z dtype)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".dtype == " false)) (dtype_program dtype))) :ruleset array_api_program_gen_ruleset) (function tuple_int_program (TupleInt) Program) -(rewrite (ndarray_program (assume_shape z ti)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".shape == " false)) (tuple_int_program ti))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (assume_isfinite z)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___init__ "assert np.all(np.isfinite(" false) (ndarray_program z)) (Program___init__ "))" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (assume_shape z ti)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert " false) (ndarray_program z)) (Program___init__ ".shape == " false)) (tuple_int_program ti))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (assume_isfinite z)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___init__ "assert np.all(np.isfinite(" false) (ndarray_program z)) (Program___init__ "))" false))) :ruleset array_api_program_gen_ruleset) (function tuple_value_program (TupleValue) Program) -(rewrite (ndarray_program (assume_value_one_of z tv)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert set(np.unique(" false) (ndarray_program z)) (Program___init__ ")) == set(" false)) (tuple_value_program tv)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(sort OptionalBool) -(function reshape (NDArray TupleInt OptionalBool) NDArray) +(rewrite (ndarray_program (assume_value_one_of z tv)) (Program_statement (ndarray_program z) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "assert set(np.unique(" false) (ndarray_program z)) (Program___init__ ")) == set(" false)) (tuple_value_program tv)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) (function Program_assign (Program) Program) -(rewrite (ndarray_program (reshape y ti ob)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".reshape(" false)) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (astype y dtype)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".astype(" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (reshape y ti ob)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".reshape(" false)) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (astype y dtype)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (ndarray_program y) (Program___init__ ".astype(" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) (function tuple_ndarray_program (TupleNDArray) Program) -(function unique_counts (NDArray) TupleNDArray) -(rewrite (tuple_ndarray_program (unique_counts x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_counts=True)" false))) :ruleset ruleset_4998548992) -(function unique_inverse (NDArray) TupleNDArray) -(rewrite (tuple_ndarray_program (unique_inverse x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_inverse=True)" false))) :ruleset ruleset_4998548992) +(rewrite (tuple_ndarray_program (unique_counts x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_counts=True)" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_ndarray_program (unique_inverse x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ", return_inverse=True)" false))) :ruleset array_api_program_gen_ruleset) (function int_program (Int) Program) -(rewrite (ndarray_program (TupleNDArray___getitem__ tnd i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program tnd) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (TupleNDArray___getitem__ tnd i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program tnd) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset array_api_program_gen_ruleset) (function value_program (Value) Program) -(rewrite (ndarray_program (NDArray_scalar v)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (value_program v)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function OptionalDType_none () OptionalDType) -(rewrite (ndarray_program (zeros ti (OptionalDType_none) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (zeros ti (OptionalDType_some dtype) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ", dtype=" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(function unique_values (NDArray) NDArray) -(rewrite (ndarray_program (unique_values x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray_scalar v)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (value_program v)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (zeros ti %__expr_2091604061025500396 optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (zeros ti (OptionalDType_some dtype) optional_device_)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.zeros(" false) (tuple_int_program ti)) (Program___init__ ", dtype=" false)) (dtype_program dtype)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (unique_values x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.unique(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) (function NDArray___add__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___add__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " + " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___sub__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " - " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___mul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " * " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___truediv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " / " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(function NDArray___lt__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___lt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " < " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___add__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " + " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___sub__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " - " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___mul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " * " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___truediv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " / " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___lt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " < " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___le__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___le__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " <= " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___gt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " > " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___le__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " <= " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___gt__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " > " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___ge__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___ge__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >= " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___eq__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " == " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___matmul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " @ " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___ge__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >= " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___eq__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " == " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___matmul__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " @ " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___mod__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___mod__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " % " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___mod__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " % " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___and__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___and__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " & " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___and__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " & " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___or__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___or__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " | " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___or__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " | " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___xor__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___xor__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ^ " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___xor__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ^ " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___lshift__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___lshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " << " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___lshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " << " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___rshift__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___rshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >> " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___rshift__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " >> " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___floordiv__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___floordiv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " // " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___floordiv__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " // " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function NDArray___pow__ (NDArray NDArray) NDArray) -(rewrite (ndarray_program (NDArray___pow__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ** " false)) (ndarray_program y))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___pow__ x y)) (Program_assign (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ " ** " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) (function index_key_program (IndexKey) Program) -(rewrite (ndarray_program (NDArray___setitem__ x idx y)) (Program_statement (Program_assign (ndarray_program x)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program_assign (ndarray_program x)) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "] = " false)) (ndarray_program y))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray___getitem__ x idx)) (Program___add__ (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (square x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.square(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (expand_dims x i)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.expand_dims(" false) (ndarray_program x)) (Program___init__ ", " false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(function mean (NDArray OptionalIntOrTuple Boolean) NDArray) -(rewrite (ndarray_program (mean x %__expr_8391745089977789030 %__expr_5801044028809884995)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (NDArray___setitem__ x idx y)) (Program_statement (Program_assign (ndarray_program x)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program_assign (ndarray_program x)) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "] = " false)) (ndarray_program y))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray___getitem__ x idx)) (Program___add__ (Program___add__ (Program___add__ (ndarray_program x) (Program___init__ "[" false)) (index_key_program idx)) (Program___init__ "]" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (square x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.square(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (expand_dims x i)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.expand_dims(" false) (ndarray_program x)) (Program___init__ ", " false)) (int_program i)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (mean x %__expr_5981948135265815040 %__expr_6185990579165616281)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) (function int_or_tuple_program (IntOrTuple) Program) -(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) %__expr_5801044028809884995)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(function TRUE () Boolean) -(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ", keepdims=True)" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (concat tnd %__expr_4873895522838289209)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (concat tnd (OptionalInt_some i))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ", axis=" false)) (int_program i)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray_vector tv)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (tuple_value_program tv)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function std (NDArray OptionalIntOrTuple) NDArray) -(rewrite (ndarray_program (std x %__expr_8391745089977789030)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (std x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (tuple_ndarray_program (svd x (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (tuple_ndarray_program (svd x %__expr_5801044028809884995)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ", full_matrices=False)" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (ndarray-sqrt x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sqrt(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (NDArray_T x)) (Program___add__ (ndarray_program x) (Program___init__ ".T" false)) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (sum x %__expr_8391745089977789030)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (sum x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) -(rewrite (tuple_int_program (NDArray_shape x)) (Program___add__ (ndarray_program x) (Program___init__ ".shape" false)) :ruleset ruleset_4998548992) -(function ndarray-abs (NDArray) NDArray) -(rewrite (ndarray_program (ndarray-abs x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset ruleset_4998548992) +(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) %__expr_6185990579165616281)) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (mean x (OptionalIntOrTuple_some int_or_tuple_) (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.mean(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ", keepdims=True)" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (concat tnd %__expr_-143818766793636568)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (concat tnd (OptionalInt_some i))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.concatenate(" false) (tuple_ndarray_program tnd)) (Program___init__ ", axis=" false)) (int_program i)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray_vector tv)) (Program___add__ (Program___add__ (Program___init__ "np.array(" false) (tuple_value_program tv)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (std x %__expr_5981948135265815040)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (std x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.std(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_ndarray_program (svd x (TRUE))) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_ndarray_program (svd x %__expr_6185990579165616281)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.linalg.svd(" false) (ndarray_program x)) (Program___init__ ", full_matrices=False)" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (ndarray-sqrt x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sqrt(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (NDArray_T x)) (Program___add__ (ndarray_program x) (Program___init__ ".T" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (sum x (OptionalIntOrTuple_none))) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (sum x (OptionalIntOrTuple_some int_or_tuple_))) (Program_assign (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "np.sum(" false) (ndarray_program x)) (Program___init__ ", axis=" false)) (int_or_tuple_program int_or_tuple_)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_int_program (NDArray_shape x)) (Program___add__ (ndarray_program x) (Program___init__ ".shape" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (ndarray-abs x)) (Program_assign (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (ndarray_program x)) (Program___init__ ")" false))) :ruleset array_api_program_gen_ruleset) (function optional_int_or_tuple_program (OptionalIntOrTuple) Program) -(rewrite (optional_int_or_tuple_program (OptionalIntOrTuple_some it)) (int_or_tuple_program it) :ruleset ruleset_4998548992) -(rewrite (optional_int_or_tuple_program %__expr_8391745089977789030) (Program___init__ "None" false) :ruleset ruleset_4998548992) -(rewrite (int_or_tuple_program (IntOrTuple_int x)) (int_program x) :ruleset ruleset_4998548992) +(rewrite (optional_int_or_tuple_program (OptionalIntOrTuple_some it)) (int_or_tuple_program it) :ruleset array_api_program_gen_ruleset) +(rewrite (optional_int_or_tuple_program %__expr_5981948135265815040) (Program___init__ "None" false) :ruleset array_api_program_gen_ruleset) +(rewrite (int_or_tuple_program (IntOrTuple_int x)) (int_program x) :ruleset array_api_program_gen_ruleset) (function IntOrTuple_tuple (TupleInt) IntOrTuple) -(rewrite (int_or_tuple_program (IntOrTuple_tuple t)) (tuple_int_program t) :ruleset ruleset_4998548992) +(rewrite (int_or_tuple_program (IntOrTuple_tuple t)) (tuple_int_program t) :ruleset array_api_program_gen_ruleset) (function IndexKey_ELLIPSIS () IndexKey) -(rewrite (index_key_program (IndexKey_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_4998548992) -(rewrite (index_key_program (IndexKey_int i)) (int_program i) :ruleset ruleset_4998548992) +(rewrite (index_key_program (IndexKey_ELLIPSIS)) (Program___init__ "..." false) :ruleset array_api_program_gen_ruleset) +(rewrite (index_key_program (IndexKey_int i)) (int_program i) :ruleset array_api_program_gen_ruleset) (function slice_program (Slice) Program) -(rewrite (index_key_program (IndexKey_slice s)) (slice_program s) :ruleset ruleset_4998548992) +(rewrite (index_key_program (IndexKey_slice s)) (slice_program s) :ruleset array_api_program_gen_ruleset) (function multi_axis_index_key_program (MultiAxisIndexKey) Program) -(rewrite (index_key_program (IndexKey_multi_axis key)) (multi_axis_index_key_program key) :ruleset ruleset_4998548992) -(rewrite (index_key_program (IndexKey_ndarray a)) (ndarray_program a) :ruleset ruleset_4998548992) +(rewrite (index_key_program (IndexKey_multi_axis key)) (multi_axis_index_key_program key) :ruleset array_api_program_gen_ruleset) +(rewrite (index_key_program (IndexKey_ndarray a)) (ndarray_program a) :ruleset array_api_program_gen_ruleset) (sort UnstableFn_MultiAxisIndexKeyItem_Int (UnstableFn (Int) MultiAxisIndexKeyItem)) -(function Int___add__ (Int Int) Int) -(function idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ (UnstableFn_MultiAxisIndexKeyItem_Int Int) MultiAxisIndexKeyItem) -(rewrite (idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_7773480733401006441)) :ruleset ruleset_4998548992) +(function idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ (UnstableFn_MultiAxisIndexKeyItem_Int Int) MultiAxisIndexKeyItem) +(rewrite (idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ idx_fn i) (unstable-app idx_fn (Int___add__ i (Int___init__ 1))) :ruleset array_api_program_gen_ruleset) (function MultiAxisIndexKey___init__ (Int UnstableFn_MultiAxisIndexKeyItem_Int) MultiAxisIndexKey) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ %__expr_871583677900651896 idx_fn)) (Program___init__ "" false) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ %__expr_-7942988945600443502 idx_fn)) (Program___init__ "" false) :ruleset array_api_program_gen_ruleset) (function multi_axis_index_key_item_program (MultiAxisIndexKeyItem) Program) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (unstable-app idx_fn %__expr_871583677900651896)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ (- k %__expr_-167787087406110176)) (unstable-fn "idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_4998548992) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_4998548992) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ "," false)) :when ((= (vec-length vec) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) -(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (vec-get vec 0)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-remove vec 0)))) :when ((> (vec-length vec) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_int i)) (int_program i) :ruleset ruleset_4998548992) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_slice s)) (slice_program s) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (unstable-app idx_fn %__expr_-7942988945600443502)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_MultiAxisIndexKeyItem_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)))) :when ((!= k 0)) :ruleset array_api_program_gen_ruleset) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-of))) (Program___init__ "" false) :ruleset array_api_program_gen_ruleset) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (multi_axis_index_key_item_program (vec-get vec %__expr_-1580149342035447281)) (Program___init__ "," false)) :when ((= (vec-length vec) %__expr_-8085216922850384227)) :ruleset array_api_program_gen_ruleset) +(rewrite (multi_axis_index_key_program (MultiAxisIndexKey_from_vec vec)) (Program___add__ (Program___add__ (multi_axis_index_key_item_program (vec-get vec %__expr_-1580149342035447281)) (Program___init__ ", " false)) (multi_axis_index_key_program (MultiAxisIndexKey_from_vec (vec-remove vec %__expr_-1580149342035447281)))) :when ((> (vec-length vec) %__expr_-8085216922850384227)) :ruleset array_api_program_gen_ruleset) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_int i)) (int_program i) :ruleset array_api_program_gen_ruleset) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_slice s)) (slice_program s) :ruleset array_api_program_gen_ruleset) (function MultiAxisIndexKeyItem_ELLIPSIS () MultiAxisIndexKeyItem) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_ELLIPSIS)) (Program___init__ "..." false) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_ELLIPSIS)) (Program___init__ "..." false) :ruleset array_api_program_gen_ruleset) (function MultiAxisIndexKeyItem_NONE () MultiAxisIndexKeyItem) -(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_NONE)) (Program___init__ "None" false) :ruleset ruleset_4998548992) +(rewrite (multi_axis_index_key_item_program (MultiAxisIndexKeyItem_NONE)) (Program___init__ "None" false) :ruleset array_api_program_gen_ruleset) (function optional_int_slice_program (OptionalInt) Program) -(rewrite (slice_program (Slice___init__ start stop %__expr_4873895522838289209)) (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) :ruleset ruleset_4998548992) -(rewrite (slice_program (Slice___init__ start stop (OptionalInt_some i))) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) (Program___init__ ":" false)) (int_program i)) :ruleset ruleset_4998548992) -(rewrite (optional_int_slice_program %__expr_4873895522838289209) (Program___init__ "" false) :ruleset ruleset_4998548992) -(rewrite (optional_int_slice_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_4998548992) +(rewrite (slice_program (Slice___init__ start stop %__expr_-143818766793636568)) (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) :ruleset array_api_program_gen_ruleset) +(rewrite (slice_program (Slice___init__ start stop (OptionalInt_some i))) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (optional_int_slice_program start) (Program___init__ ":" false)) (optional_int_slice_program stop)) (Program___init__ ":" false)) (int_program i)) :ruleset array_api_program_gen_ruleset) +(rewrite (optional_int_slice_program %__expr_-143818766793636568) (Program___init__ "" false) :ruleset array_api_program_gen_ruleset) +(rewrite (optional_int_slice_program (OptionalInt_some x)) (int_program x) :ruleset array_api_program_gen_ruleset) (function optional_int_program (OptionalInt) Program) -(rewrite (optional_int_program %__expr_4873895522838289209) (Program___init__ "None" false) :ruleset ruleset_4998548992) -(rewrite (optional_int_program (OptionalInt_some x)) (int_program x) :ruleset ruleset_4998548992) +(rewrite (optional_int_program %__expr_-143818766793636568) (Program___init__ "None" false) :ruleset array_api_program_gen_ruleset) +(rewrite (optional_int_program (OptionalInt_some x)) (int_program x) :ruleset array_api_program_gen_ruleset) (function optional_dtype_program (OptionalDType) Program) -(rewrite (optional_dtype_program (OptionalDType_none)) (Program___init__ "None" false) :ruleset ruleset_4998548992) -(rewrite (optional_dtype_program (OptionalDType_some dtype)) (dtype_program dtype) :ruleset ruleset_4998548992) +(rewrite (optional_dtype_program %__expr_2091604061025500396) (Program___init__ "None" false) :ruleset array_api_program_gen_ruleset) +(rewrite (optional_dtype_program (OptionalDType_some dtype)) (dtype_program dtype) :ruleset array_api_program_gen_ruleset) (function tuple_ndarray_program_inner (TupleNDArray) Program) -(rewrite (tuple_ndarray_program r) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_ndarray_program_inner r)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(rewrite (tuple_ndarray_program_inner (TupleNDArray___init__ x)) (Program___add__ (ndarray_program x) (Program___init__ "," false)) :ruleset ruleset_4998548992) -(rewrite (tuple_ndarray_program_inner (TupleNDArray___add__ l r)) (Program___add__ (Program___add__ (tuple_ndarray_program_inner l) (Program___init__ " " false)) (tuple_ndarray_program_inner r)) :ruleset ruleset_4998548992) -(function TupleNDArray_length (TupleNDArray) Int) -(rewrite (int_program (TupleNDArray_length l)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_ndarray_program l)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(rewrite (ndarray_program (TupleNDArray___getitem__ l i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program l) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) +(rewrite (tuple_ndarray_program r) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_ndarray_program_inner r)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_ndarray_program_inner (TupleNDArray___init__ x)) (Program___add__ (ndarray_program x) (Program___init__ "," false)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_ndarray_program_inner (TupleNDArray___add__ l r)) (Program___add__ (Program___add__ (tuple_ndarray_program_inner l) (Program___init__ " " false)) (tuple_ndarray_program_inner r)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (TupleNDArray_length l)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_ndarray_program l)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (ndarray_program (TupleNDArray___getitem__ l i)) (Program___add__ (Program___add__ (Program___add__ (tuple_ndarray_program l) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset array_api_program_gen_ruleset) (function tuple_value_program_inner (TupleValue) Program) -(rewrite (tuple_value_program tv1) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_value_program_inner tv1)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(rewrite (tuple_value_program_inner (TupleValue___add__ tv1 tv2)) (Program___add__ (Program___add__ (tuple_value_program_inner tv1) (Program___init__ " " false)) (tuple_value_program_inner tv2)) :ruleset ruleset_4998548992) -(rewrite (tuple_value_program_inner (TupleValue___init__ v)) (Program___add__ (value_program v) (Program___init__ "," false)) :ruleset ruleset_4998548992) -(rewrite (value_program (Value_int i)) (int_program i) :ruleset ruleset_4998548992) -(function Value_bool (Boolean) Value) +(rewrite (tuple_value_program tv1) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_value_program_inner tv1)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_value_program_inner (TupleValue___add__ tv1 tv2)) (Program___add__ (Program___add__ (tuple_value_program_inner tv1) (Program___init__ " " false)) (tuple_value_program_inner tv2)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_value_program_inner (TupleValue___init__ v)) (Program___add__ (value_program v) (Program___init__ "," false)) :ruleset array_api_program_gen_ruleset) +(rewrite (value_program (Value_int i)) (int_program i) :ruleset array_api_program_gen_ruleset) (function bool_program (Boolean) Program) -(rewrite (value_program (Value_bool b)) (bool_program b) :ruleset ruleset_4998548992) +(rewrite (value_program (Value_bool b)) (bool_program b) :ruleset array_api_program_gen_ruleset) (function float_program (Float) Program) -(rewrite (value_program (Value_float f)) (float_program f) :ruleset ruleset_4998548992) -(rewrite (value_program (NDArray_to_value x)) (ndarray_program x) :ruleset ruleset_4998548992) -(function Value___lt__ (Value Value) Value) -(rewrite (value_program (Value___lt__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " < " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Value___truediv__ (Value Value) Value) -(rewrite (value_program (Value___truediv__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " / " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Value_to_bool (Value) Boolean) -(rewrite (bool_program (Value_to_bool v1)) (value_program v1) :ruleset ruleset_4998548992) -(rewrite (int_program (Value_to_int v1)) (value_program v1) :ruleset ruleset_4998548992) -(rewrite (float_program (Float___init__ f64_)) (Program___init__ (to-string f64_) false) :ruleset ruleset_4998548992) -(function Float_abs (Float) Float) -(rewrite (float_program (Float_abs f)) (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (float_program f)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Float_from_int (Int) Float) -(rewrite (float_program (Float_from_int i)) (int_program i) :ruleset ruleset_4998548992) -(function Float___add__ (Float Float) Float) -(rewrite (float_program (Float___add__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " + " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Float___sub__ (Float Float) Float) -(rewrite (float_program (Float___sub__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " - " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Float___mul__ (Float Float) Float) -(rewrite (float_program (Float___mul__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " * " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Float___truediv__ (Float Float) Float) -(rewrite (float_program (Float___truediv__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " / " false)) (float_program g)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ " / " false)) (Program___init__ (to-string (denom r)) false)) (Program___init__ ")" false)) :when ((!= (denom r) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) -(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ ")" false)) :when ((= (denom r) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) -(rewrite (dtype_program %__expr_-458881301673629543) (Program___init__ "np.dtype(np.float64)" false) :ruleset ruleset_4998548992) -(function DType_float32 () DType) -(rewrite (dtype_program (DType_float32)) (Program___init__ "np.dtype(np.float32)" false) :ruleset ruleset_4998548992) -(rewrite (dtype_program (DType_int64)) (Program___init__ "np.dtype(np.int64)" false) :ruleset ruleset_4998548992) -(rewrite (dtype_program %__expr_7346177422650294070) (Program___init__ "np.dtype(np.int32)" false) :ruleset ruleset_4998548992) -(function DType_bool () DType) -(rewrite (dtype_program (DType_bool)) (Program___init__ "np.dtype(np.bool)" false) :ruleset ruleset_4998548992) -(function DType_object () DType) -(rewrite (dtype_program (DType_object)) (Program___init__ "np.dtype(np.object_)" false) :ruleset ruleset_4998548992) -(function Program_function_two (Program Program Program String) Program) -(function Program_compile (Program i64) Unit :default ()) -(rule ((= f (ndarray_function_two res l r))) - ((union f (Program_function_two (ndarray_program res) (ndarray_program l) (ndarray_program r) "__fn")) - (Program_compile f 0)) - :ruleset ruleset_4998548992 ) -(sort UnstableFn_Int_Int (UnstableFn (Int) Int)) -(function idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ (UnstableFn_Int_Int Int) Int) -(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__ idx_fn i) (unstable-app idx_fn (Int___add__ i %__expr_7773480733401006441)) :ruleset ruleset_4998548992) -(rewrite (int_program (TupleInt___getitem__ ti i)) (Program___add__ (Program___add__ (Program___add__ (tuple_int_program ti) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset ruleset_4998548992) -(function TupleInt_length (TupleInt) Int) -(rewrite (int_program (TupleInt_length ti)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_int_program ti)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (value_program (Value_float f)) (float_program f) :ruleset array_api_program_gen_ruleset) +(rewrite (value_program (NDArray_to_value x)) (ndarray_program x) :ruleset array_api_program_gen_ruleset) +(rewrite (value_program (Value___lt__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " < " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (value_program (Value___truediv__ v1 v2)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (value_program v1)) (Program___init__ " / " false)) (value_program v2)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program (Value_to_bool v1)) (value_program v1) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Value_to_int v1)) (value_program v1) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float___init__ f64_)) (Program___init__ (to-string f64_) false) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float_abs f)) (Program___add__ (Program___add__ (Program___init__ "np.abs(" false) (float_program f)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float_from_int i)) (int_program i) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float___add__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " + " false)) (float_program g)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float___sub__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " - " false)) (float_program g)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float___mul__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " * " false)) (float_program g)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float___truediv__ f g)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (float_program f)) (Program___init__ " / " false)) (float_program g)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ " / " false)) (Program___init__ (to-string (denom r)) false)) (Program___init__ ")" false)) :when ((!= (denom r) %__expr_-8085216922850384227)) :ruleset array_api_program_gen_ruleset) +(rewrite (float_program (Float_rational r)) (Program___add__ (Program___add__ (Program___init__ "float(" false) (Program___init__ (to-string (numer r)) false)) (Program___init__ ")" false)) :when ((= (denom r) %__expr_-8085216922850384227)) :ruleset array_api_program_gen_ruleset) +(rewrite (dtype_program %__expr_2565664028803046192) (Program___init__ "np.dtype(np.float64)" false) :ruleset array_api_program_gen_ruleset) +(rewrite (dtype_program (DType_float32)) (Program___init__ "np.dtype(np.float32)" false) :ruleset array_api_program_gen_ruleset) +(rewrite (dtype_program (DType_int64)) (Program___init__ "np.dtype(np.int64)" false) :ruleset array_api_program_gen_ruleset) +(rewrite (dtype_program %__expr_-3501279694271260104) (Program___init__ "np.dtype(np.int32)" false) :ruleset array_api_program_gen_ruleset) +(rewrite (dtype_program (DType_bool)) (Program___init__ "np.dtype(np.bool)" false) :ruleset array_api_program_gen_ruleset) +(rewrite (dtype_program (DType_object)) (Program___init__ "np.dtype(np.object_)" false) :ruleset array_api_program_gen_ruleset) +(rewrite (idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___ idx_fn i) (unstable-app idx_fn (Int___add__ i (Int___init__ 1))) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (TupleInt___getitem__ ti i)) (Program___add__ (Program___add__ (Program___add__ (tuple_int_program ti) (Program___init__ "[" false)) (int_program i)) (Program___init__ "]" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (TupleInt_length ti)) (Program___add__ (Program___add__ (Program___init__ "len(" false) (tuple_int_program ti)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) (function tuple_int_program_inner (TupleInt) Program) -(rewrite (tuple_int_program ti) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_int_program_inner ti)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function TupleInt___init__ (Int UnstableFn_Int_Int) TupleInt :unextractable) -(rewrite (tuple_int_program_inner (TupleInt___init__ %__expr_871583677900651896 idx_fn)) (Program___init__ "" false) :ruleset ruleset_4998548992) -(rewrite (tuple_int_program_inner (TupleInt___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (int_program (unstable-app idx_fn %__expr_871583677900651896)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt___init__ (Int___init__ (- k %__expr_-167787087406110176)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i_%__expr_7773480733401006441__" idx_fn)))) :when ((!= k 0)) :ruleset ruleset_4998548992) -(rewrite (tuple_int_program_inner (TupleInt_from_vec (vec-of))) (Program___init__ "" false) :ruleset ruleset_4998548992) -(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt_from_vec (vec-remove vec_int 0)))) :when ((> (vec-length vec_int) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) -(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (int_program (vec-get vec_int 0)) (Program___init__ "," false)) :when ((= (vec-length vec_int) %__expr_-167787087406110176)) :ruleset ruleset_4998548992) -(rewrite (int_program (Int___init__ i64_)) (Program___init__ (to-string i64_) false) :ruleset ruleset_4998548992) -(function Int___invert__ (Int) Int) -(rewrite (int_program (Int___invert__ i)) (Program___add__ (Program___init__ "~" false) (int_program i)) :ruleset ruleset_4998548992) -(function Int___lt__ (Int Int) Boolean) -(rewrite (bool_program (Int___lt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " < " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (tuple_int_program ti) (Program___add__ (Program___add__ (Program___init__ "(" false) (tuple_int_program_inner ti)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_int_program_inner (TupleInt___init__ (Int___init__ 0) idx_fn)) (Program___init__ "" false) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_int_program_inner (TupleInt___init__ (Int___init__ k) idx_fn)) (Program___add__ (Program___add__ (int_program (unstable-app idx_fn (Int___init__ 0))) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt___init__ (Int___init__ (- k 1)) (unstable-fn "idx_fn-UnstableFn_Int_Int_i-Int__unstable-app_idx_fn__Int___add___i__Int___init___1___" idx_fn)))) :when ((!= k 0)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_int_program_inner (TupleInt_from_vec (vec-of))) (Program___init__ "" false) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (Program___add__ (int_program (vec-get vec_int %__expr_-1580149342035447281)) (Program___init__ ", " false)) (tuple_int_program_inner (TupleInt_from_vec (vec-remove vec_int %__expr_-1580149342035447281)))) :when ((> (vec-length vec_int) %__expr_-8085216922850384227)) :ruleset array_api_program_gen_ruleset) +(rewrite (tuple_int_program_inner (TupleInt_from_vec vec_int)) (Program___add__ (int_program (vec-get vec_int %__expr_-1580149342035447281)) (Program___init__ "," false)) :when ((= (vec-length vec_int) %__expr_-8085216922850384227)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___init__ i64_)) (Program___init__ (to-string i64_) false) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___invert__ i)) (Program___add__ (Program___init__ "~" false) (int_program i)) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program (Int___lt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " < " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) (function Int___le__ (Int Int) Boolean) -(rewrite (bool_program (Int___le__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " <= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___gt__ (Int Int) Boolean) -(rewrite (bool_program (Int___gt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " > " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___ge__ (Int Int) Boolean) -(rewrite (bool_program (Int___ge__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___eq__ (Int Int) Boolean) -(rewrite (bool_program (Int___eq__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " == " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(rewrite (int_program (Int___add__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " + " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___sub__ (Int Int) Int) -(rewrite (int_program (Int___sub__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " - " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___mul__ (Int Int) Int) -(rewrite (int_program (Int___mul__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " * " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___truediv__ (Int Int) Int) -(rewrite (int_program (Int___truediv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " / " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___mod__ (Int Int) Int) -(rewrite (int_program (Int___mod__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " % " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (bool_program (Int___le__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " <= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program (Int___gt__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " > " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program (Int___ge__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >= " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program (Int___eq__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " == " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___add__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " + " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___sub__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " - " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___mul__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " * " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___truediv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " / " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___mod__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " % " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) (function Int___pow__ (Int Int) Int) -(rewrite (int_program (Int___pow__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ** " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___and__ (Int Int) Int) -(rewrite (int_program (Int___and__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " & " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___or__ (Int Int) Int) -(rewrite (int_program (Int___or__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " | " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___xor__ (Int Int) Int) -(rewrite (int_program (Int___xor__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ^ " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___lshift__ (Int Int) Int) -(rewrite (int_program (Int___lshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " << " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(function Int___rshift__ (Int Int) Int) -(rewrite (int_program (Int___rshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >> " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) +(rewrite (int_program (Int___pow__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ** " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___and__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " & " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___or__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " | " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___xor__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " ^ " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___lshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " << " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (int_program (Int___rshift__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " >> " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) (function Int___floordiv__ (Int Int) Int) -(rewrite (int_program (Int___floordiv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " // " false)) (int_program j)) (Program___init__ ")" false)) :ruleset ruleset_4998548992) -(rewrite (bool_program (TRUE)) (Program___init__ "True" false) :ruleset ruleset_4998548992) -(rewrite (bool_program %__expr_5801044028809884995) (Program___init__ "False" false) :ruleset ruleset_4998548992) -(ruleset ruleset_4997969680) +(rewrite (int_program (Int___floordiv__ i j)) (Program___add__ (Program___add__ (Program___add__ (Program___add__ (Program___init__ "(" false) (int_program i)) (Program___init__ " // " false)) (int_program j)) (Program___init__ ")" false)) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program (TRUE)) (Program___init__ "True" false) :ruleset array_api_program_gen_ruleset) +(rewrite (bool_program %__expr_6185990579165616281) (Program___init__ "False" false) :ruleset array_api_program_gen_ruleset) +(ruleset program_gen_ruleset) (function Program_expr (Program) String) (function Program_statements (Program) String) (function Program_next_sym (Program) i64) @@ -1550,20 +1378,20 @@ (set (Program_statements p) "") (set (Program_next_sym p) i) (set (Program_is_identifer p) b)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (function Program_expr_to_statement (Program) Program) -(rewrite (Program_statement p1 p2) (Program___add__ p1 (Program_expr_to_statement p2)) :ruleset ruleset_4997969680) +(rewrite (Program_statement p1 p2) (Program___add__ p1 (Program_expr_to_statement p2)) :ruleset program_gen_ruleset) (function Program_parent (Program) Program :unextractable :merge old) (rule ((= p (Program_expr_to_statement p1)) (Program_compile p i)) ((set (Program_parent p1) p) (set (Program_is_identifer p) false)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_expr_to_statement p1)) (Program_compile p i) (= (Program_parent p1) p)) ((Program_compile p1 i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_expr_to_statement p1)) (Program_compile p i) (!= (Program_parent p1) p) @@ -1572,7 +1400,7 @@ ")) (set (Program_next_sym p) i) (set (Program_expr p) "")) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_expr_to_statement p1)) (= (Program_parent p1) p) (= s1 (Program_expr p1)) @@ -1582,54 +1410,54 @@ ")) (set (Program_next_sym p) i) (set (Program_expr p) "")) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (= (Program_expr p) (Program_expr p1)) (= b (Program_is_identifer p1))) ((set (Program_is_identifer p) b)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (= (Program_expr p) (Program_expr p2)) (= b (Program_is_identifer p2))) ((set (Program_is_identifer p) b)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (!= (Program_expr p) (Program_expr p1)) (!= (Program_expr p) (Program_expr p2))) ((set (Program_is_identifer p) false)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i)) ((set (Program_parent p1) p)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (= (Program_parent p1) p)) ((Program_compile p1 i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (Program_next_sym p1)) ((set (Program_parent p2) p)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (!= (Program_parent p1) p) (= (Program_parent p2) p)) ((Program_compile p2 i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i2) (= (Program_parent p1) p) (= i (Program_next_sym p1)) (= (Program_parent p2) p)) ((Program_compile p2 i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (= s1 (Program_expr p1)) (= s2 (Program_expr p2))) ((set (Program_expr p) (+ s1 s2))) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (= (Program_parent p1) p) (= (Program_parent p2) p) @@ -1638,14 +1466,14 @@ (= i (Program_next_sym p2))) ((set (Program_statements p) (+ s1 s2)) (set (Program_next_sym p) i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (Program_compile p i) (!= (Program_parent p1) p) (!= (Program_parent p2) p)) ((set (Program_statements p) "") (set (Program_next_sym p) i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (= (Program_parent p1) p) (!= (Program_parent p2) p) @@ -1653,7 +1481,7 @@ (= i (Program_next_sym p1))) ((set (Program_statements p) s1) (set (Program_next_sym p) i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program___add__ p1 p2)) (= (Program_parent p2) p) (!= (Program_parent p1) p) @@ -1661,17 +1489,17 @@ (= i (Program_next_sym p2))) ((set (Program_statements p) s2) (set (Program_next_sym p) i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_assign p1)) (Program_compile p i)) ((set (Program_parent p1) p) (set (Program_is_identifer p) true)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_assign p1)) (Program_compile p i) (= (Program_parent p1) p)) ((Program_compile p1 i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_assign p1)) (= (Program_parent p1) p) (= s1 (Program_statements p1)) @@ -1681,8 +1509,8 @@ ((set (Program_statements p) (+ s1 (+ "_" (to-string i)) " = " s2 " ")) (set (Program_expr p) (+ "_" (to-string i))) - (set (Program_next_sym p) (+ i %__expr_-167787087406110176))) - :ruleset ruleset_4997969680 ) + (set (Program_next_sym p) (+ i %__expr_-8085216922850384227))) + :ruleset program_gen_ruleset ) (rule ((= p (Program_assign p1)) (!= (Program_parent p1) p) (Program_compile p i) @@ -1691,8 +1519,8 @@ ((set (Program_statements p) (+ (+ "_" (to-string i)) " = " s2 " ")) (set (Program_expr p) (+ "_" (to-string i))) - (set (Program_next_sym p) (+ i %__expr_-167787087406110176))) - :ruleset ruleset_4997969680 ) + (set (Program_next_sym p) (+ i %__expr_-8085216922850384227))) + :ruleset program_gen_ruleset ) (rule ((= p (Program_assign p1)) (= (Program_parent p1) p) (= s1 (Program_statements p1)) @@ -1702,7 +1530,7 @@ ((set (Program_statements p) s1) (set (Program_expr p) s2) (set (Program_next_sym p) i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_assign p1)) (!= (Program_parent p1) p) (Program_compile p i) @@ -1711,7 +1539,7 @@ ((set (Program_statements p) "") (set (Program_expr p) s2) (set (Program_next_sym p) i)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_function_two p1 p2 p3 s1)) (Program_compile p i)) ((set (Program_parent p2) p) @@ -1721,7 +1549,7 @@ (Program_compile p3 i) (Program_compile p1 i) (set (Program_is_identifer p) true)) - :ruleset ruleset_4997969680 ) + :ruleset program_gen_ruleset ) (rule ((= p (Program_function_two p1 p2 p3 s1)) (Program_compile p i) (= s2 (Program_expr p1)) @@ -1735,6 +1563,7 @@ ")) (set (Program_next_sym p) i) (set (Program_expr p) s1)) - :ruleset ruleset_4997969680 ) -(run-schedule (seq (saturate (run ruleset_4998548992)) (saturate (run ruleset_4997969680)))) -(extract (Program_statements %fn_program) 0) + :ruleset program_gen_ruleset ) +(run-schedule (seq (saturate (run array_api_program_gen_ruleset)) (saturate (run program_gen_ruleset)))) +(let %__expr_-116196252424162806 (assume_dtype (NDArray_var "X") (DType_float64))) +(extract (Program_statements (ndarray_function_two_program (NDArray___getitem__ (NDArray___matmul__ (NDArray___sub__ %__expr_978886785202701864 %__expr_4447619498771141281) (NDArray___matmul__ %__expr_-6562228218073754698 (NDArray___getitem__ (NDArray_T (TupleNDArray___getitem__ %__expr_-4464433185064523998 %__expr_2305922367412063525)) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some (Value_to_int (NDArray_to_value (sum (astype (NDArray___gt__ %__expr_2615935926139764667 (NDArray___mul__ %__expr_-4180228001421012884 (NDArray___getitem__ %__expr_2615935926139764667 %__expr_4326809257670680665))) %__expr_-3501279694271260104) %__expr_5981948135265815040)))) %__expr_-143818766793636568)))))))) (IndexKey_multi_axis (MultiAxisIndexKey_from_vec (vec-of %__expr_-8637348939342624324 (MultiAxisIndexKeyItem_slice (Slice___init__ %__expr_-143818766793636568 (OptionalInt_some %__expr_2305922367412063525) %__expr_-143818766793636568)))))) (assume_isfinite (assume_shape (assume_dtype (NDArray_var "X") (DType_float64)) (TupleInt_from_vec (vec-of (Int___init__ 150) (Int___init__ 4))))) (assume_value_one_of (assume_shape (assume_dtype (NDArray_var "y") (DType_int64)) (TupleInt_from_vec (vec-of (Int___init__ 150)))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 0))) (TupleValue___add__ (TupleValue___init__ (Value_int (Int___init__ 1))) (TupleValue___init__ (Value_int (Int___init__ 2)))))))) 0) From c537b79e93a895cbfd35b08ef4d0bc8f36af8fc3 Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 24 Oct 2024 10:57:35 -0400 Subject: [PATCH 08/11] rename big python file --- .../{python_array_optimize_big.egg => stresstest_large_expr.egg} | 0 1 file changed, 0 insertions(+), 0 deletions(-) rename tests/{python_array_optimize_big.egg => stresstest_large_expr.egg} (100%) diff --git a/tests/python_array_optimize_big.egg b/tests/stresstest_large_expr.egg similarity index 100% rename from tests/python_array_optimize_big.egg rename to tests/stresstest_large_expr.egg From 32e55e1ed31a77f5637bd49ecd7a66de75733e3b Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 24 Oct 2024 10:59:47 -0400 Subject: [PATCH 09/11] make it clear its performance thats non deterministic --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d2eb3a84..9e663065 100644 --- a/README.md +++ b/README.md @@ -133,8 +133,9 @@ capture a single trace of the CPU interactions ([src](https://docs.codspeed.io/f > CodSpeed instruments your benchmarks to measure the performance of your code. A benchmark will be run only once and the CPU behavior will be simulated. This ensures that the measurement is as accurate as possible, taking into account not only the instructions executed but also the cache and memory access patterns. The simulation gives us an equivalent of the CPU cycles that includes cache and memory access. -Since many of the shorter running benchmarks have unstable timings due to non deterministic behavior, we ["ignore"](https://docs.codspeed.io/features/ignoring-benchmarks/) -them in codspeed. That way, we still capture their performance, but their timings don't show up in our reports by default. +Since many of the shorter running benchmarks have unstable timings due to non deterministic performance ([like in the memory allocator](https://github.com/oxc-project/backlog/issues/89)), +we ["ignore"](https://docs.codspeed.io/features/ignoring-benchmarks/) them in codspeed. That way, we still +capture their performance, but their timings don't show up in our reports by default. We use 50ms as our cutoff currently, any benchmarks shorter than that are ignored. This number was selected to try to ignore any benchmarks with have changes > 1% when they haven't been modified. From eb4a836fd496c3b4c06de0d6cb7add46655db47e Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 24 Oct 2024 11:00:59 -0400 Subject: [PATCH 10/11] Clarify ignoring benchmarks is manual --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 9e663065..552f66b1 100644 --- a/README.md +++ b/README.md @@ -138,4 +138,5 @@ we ["ignore"](https://docs.codspeed.io/features/ignoring-benchmarks/) them in co capture their performance, but their timings don't show up in our reports by default. We use 50ms as our cutoff currently, any benchmarks shorter than that are ignored. This number was selected to try to ignore -any benchmarks with have changes > 1% when they haven't been modified. +any benchmarks with have changes > 1% when they haven't been modified. Note that all the ignoring is done manually, +so if you add another example that's short, an admin on the codspeed project will need to manually ignore it. From 00820057388469a82e7388a81128402974c89a7c Mon Sep 17 00:00:00 2001 From: Saul Shanabrook Date: Thu, 24 Oct 2024 11:01:30 -0400 Subject: [PATCH 11/11] Move benchmarks above docs --- README.md | 32 ++++++++++++++++---------------- 1 file changed, 16 insertions(+), 16 deletions(-) diff --git a/README.md b/README.md index 552f66b1..7b69116e 100644 --- a/README.md +++ b/README.md @@ -57,6 +57,22 @@ for the REPL. To run the tests use `make test`. +## Benchmarks + +We run all of our "examples" [as benchmarks in codspeed](https://codspeed.io/egraphs-good/egglog). These are in CI +for every commit in main and for all PRs. It will run the examples with extra instrumentation added so that it can +capture a single trace of the CPU interactions ([src](https://docs.codspeed.io/features/understanding-the-metrics/)): + +> CodSpeed instruments your benchmarks to measure the performance of your code. A benchmark will be run only once and the CPU behavior will be simulated. This ensures that the measurement is as accurate as possible, taking into account not only the instructions executed but also the cache and memory access patterns. The simulation gives us an equivalent of the CPU cycles that includes cache and memory access. + +Since many of the shorter running benchmarks have unstable timings due to non deterministic performance ([like in the memory allocator](https://github.com/oxc-project/backlog/issues/89)), +we ["ignore"](https://docs.codspeed.io/features/ignoring-benchmarks/) them in codspeed. That way, we still +capture their performance, but their timings don't show up in our reports by default. + +We use 50ms as our cutoff currently, any benchmarks shorter than that are ignored. This number was selected to try to ignore +any benchmarks with have changes > 1% when they haven't been modified. Note that all the ignoring is done manually, +so if you add another example that's short, an admin on the codspeed project will need to manually ignore it. + # Documentation To view documentation, run `cargo doc --open`. @@ -124,19 +140,3 @@ These primitives are only defined when the result itself is a pure rational. Use double quotes to get a quote: `"Foo "" Bar"` is `Foo " Bar`. No primitives defined. - -## Benchmarks - -We run all of our "examples" [as benchmarks in codspeed](https://codspeed.io/egraphs-good/egglog). These are in CI -for every commit in main and for all PRs. It will run the examples with extra instrumentation added so that it can -capture a single trace of the CPU interactions ([src](https://docs.codspeed.io/features/understanding-the-metrics/)): - -> CodSpeed instruments your benchmarks to measure the performance of your code. A benchmark will be run only once and the CPU behavior will be simulated. This ensures that the measurement is as accurate as possible, taking into account not only the instructions executed but also the cache and memory access patterns. The simulation gives us an equivalent of the CPU cycles that includes cache and memory access. - -Since many of the shorter running benchmarks have unstable timings due to non deterministic performance ([like in the memory allocator](https://github.com/oxc-project/backlog/issues/89)), -we ["ignore"](https://docs.codspeed.io/features/ignoring-benchmarks/) them in codspeed. That way, we still -capture their performance, but their timings don't show up in our reports by default. - -We use 50ms as our cutoff currently, any benchmarks shorter than that are ignored. This number was selected to try to ignore -any benchmarks with have changes > 1% when they haven't been modified. Note that all the ignoring is done manually, -so if you add another example that's short, an admin on the codspeed project will need to manually ignore it.