diff --git a/Cargo.lock b/Cargo.lock index 545b6b94..32dccb7e 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -130,6 +130,15 @@ version = "2.5.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "cf4b9d6a944f767f8e5e0db018570623c85f3d925ac718db4e06d0187adb21c1" +[[package]] +name = "bitmaps" +version = "2.1.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "031043d04099746d8db04daf1fa424b2bc8bd69d92b25962dcde24da39ab64a2" +dependencies = [ + "typenum", +] + [[package]] name = "block-buffer" version = "0.10.4" @@ -295,6 +304,7 @@ dependencies = [ "getrandom", "glob", "hashbrown 0.14.3", + "im-rc", "indexmap", "instant", "lalrpop", @@ -472,6 +482,20 @@ version = "2.1.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "9a3a5bfb195931eeb336b2a7b4d761daec841b97f947d34394601737a7bba5e4" +[[package]] +name = "im-rc" +version = "15.1.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "af1955a75fa080c677d3972822ec4bad316169ab1cfc6c257a942c2265dbe5fe" +dependencies = [ + "bitmaps", + "rand_core", + "rand_xoshiro", + "sized-chunks", + "typenum", + "version_check", +] + [[package]] name = "indexmap" version = "2.2.6" @@ -880,6 +904,15 @@ dependencies = [ "serde", ] +[[package]] +name = "rand_xoshiro" +version = "0.6.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6f97cdb2a36ed4183de61b2f824cc45c9f1037f28afe0a322e9fff4c108b5aaa" +dependencies = [ + "rand_core", +] + [[package]] name = "redox_syscall" version = "0.4.1" @@ -1024,6 +1057,16 @@ version = "0.3.11" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "38b58827f4464d87d377d175e90bf58eb00fd8716ff0a62f80356b5e61555d0d" +[[package]] +name = "sized-chunks" +version = "0.6.5" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "16d69225bde7a69b235da73377861095455d298f2b970996eec25ddbb42b3d1e" +dependencies = [ + "bitmaps", + "typenum", +] + [[package]] name = "smallvec" version = "1.13.2" diff --git a/Cargo.toml b/Cargo.toml index 609b3b38..81b9ecf8 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -54,6 +54,9 @@ ordered-float = { version = "3.7" } # Need to add "js" feature for "graphviz-rust" to work in wasm getrandom = { version = "0.2.10", features = ["js"], optional = true } +im-rc = "15.1.0" + + [build-dependencies] lalrpop = "0.20" diff --git a/src/constraint.rs b/src/constraint.rs index 8c94f888..a29edf03 100644 --- a/src/constraint.rs +++ b/src/constraint.rs @@ -8,10 +8,13 @@ use crate::{ }, sort::I64Sort, typechecking::TypeError, - util::{FreshGen, HashMap, HashSet, SymbolGen}, + util::{FreshGen, HashSet, SymbolGen}, ArcSort, CorrespondingVar, Span, Symbol, TypeInfo, DUMMY_SPAN, }; use core::hash::Hash; +// Use immutable hashmap for performance +// cloning assignments is common and O(1) with immutable hashmap +use im_rc::HashMap; use std::{fmt::Debug, iter::once, mem::swap}; #[derive(Clone, Debug)] @@ -198,12 +201,13 @@ impl Default for Problem { } } -#[derive(Clone, Debug)] +#[derive(Clone)] pub(crate) struct Assignment(pub HashMap); impl Assignment where - Var: Hash + Eq + PartialEq, + Var: Hash + Eq + PartialEq + Clone, + Value: Clone, { pub fn insert(&mut self, var: Var, value: Value) -> Option { self.0.insert(var, value) diff --git a/src/serialize.rs b/src/serialize.rs index 603011c4..c15ee7f7 100644 --- a/src/serialize.rs +++ b/src/serialize.rs @@ -43,6 +43,7 @@ impl EGraph { /// - Primitives: 1.0 /// - Function without costs: 1.0 /// - Function with costs: the cost + /// - Omitted nodes: infinite /// /// For node IDs: /// - Functions: Function name + hash of input values @@ -127,8 +128,7 @@ impl EGraph { .0; let children: Vec<_> = input .iter() - // Filter out children which don't have an ID, meaning that we skipped emitting them due to size constraints - .filter_map(|v| self.serialize_value(&mut egraph, &mut node_ids, v, None).1) + .map(|v| self.serialize_value(&mut egraph, &mut node_ids, v, None).1) .collect(); egraph.nodes.insert( node_id, @@ -141,12 +141,11 @@ impl EGraph { ); } - let roots = config + egraph.root_eclasses = config .root_eclasses .iter() .map(|v| self.serialize_value(&mut egraph, &mut node_ids, v, None).0) .collect(); - egraph.root_eclasses = roots; egraph } @@ -163,14 +162,14 @@ impl EGraph { // The node ID to use for a primitive value, if this is None, use the hash of the value and the sort name // Set iff `split_primitive_outputs` is set and this is an output of a function. prim_node_id: Option, - ) -> (egraph_serialize::ClassId, Option) { + ) -> (egraph_serialize::ClassId, egraph_serialize::NodeId) { let sort = self.get_sort_from_value(value).unwrap(); - let (class_id, node_id): (egraph_serialize::ClassId, Option) = + let (class_id, node_id): (egraph_serialize::ClassId, egraph_serialize::NodeId) = if sort.is_eq_sort() { let id: usize = value.bits as usize; let canonical: usize = self.unionfind.find(Id::from(id)).into(); let class_id: egraph_serialize::ClassId = canonical.to_string().into(); - (class_id.clone(), get_node_id(node_ids, class_id)) + (class_id.clone(), get_node_id(egraph, node_ids, class_id)) } else { let (class_id, node_id): (egraph_serialize::ClassId, egraph_serialize::NodeId) = if let Some(node_id) = prim_node_id { @@ -183,14 +182,14 @@ impl EGraph { }; // Add node for value { + // Children will be empty unless this is a container sort let children: Vec = sort .inner_values(value) .into_iter() - .filter_map(|(_, v)| self.serialize_value(egraph, node_ids, &v, None).1) + .map(|(_, v)| self.serialize_value(egraph, node_ids, &v, None).1) .collect(); // If this is a container sort, use the name, otherwise use the value - let op: String = if sort.is_container_sort() { - log::warn!("{} is a container sort", sort.name()); + let op = if sort.is_container_sort() { sort.serialized_name(value).to_string() } else { sort.make_expr(self, *value).1.to_string() @@ -205,7 +204,7 @@ impl EGraph { }, ); }; - (class_id, Some(node_id)) + (class_id, node_id) }; egraph.class_data.insert( class_id.clone(), @@ -221,10 +220,25 @@ type NodeIDs = HashMap>, class_id: egraph_serialize::ClassId, -) -> Option { - let node_ids = node_ids.get_mut(&class_id)?; +) -> egraph_serialize::NodeId { + // If we don't find node IDs for this class, it means that all nodes for it were omitted due to size constraints + // In this case, add a dummy node in this class to represent the missing nodes + let node_ids = node_ids.entry(class_id.clone()).or_insert_with(|| { + let node_id = egraph_serialize::NodeId::from(format!("{}-dummy", class_id)); + egraph.nodes.insert( + node_id.clone(), + egraph_serialize::Node { + op: "[...]".to_string(), + eclass: class_id.clone(), + cost: NotNan::new(f64::INFINITY).unwrap(), + children: vec![], + }, + ); + VecDeque::from(vec![node_id]) + }); node_ids.rotate_left(1); - Some(node_ids.front().unwrap().clone()) + node_ids.front().unwrap().clone() } diff --git a/tests/python_array_optimize.egg b/tests/python_array_optimize.egg new file mode 100644 index 00000000..7be04f5d --- /dev/null +++ b/tests/python_array_optimize.egg @@ -0,0 +1,438 @@ +(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))))))))